Diff of /main.py [000000] .. [1bdb11]

Switch to side-by-side view

--- a
+++ b/main.py
@@ -0,0 +1,42 @@
+"""
+Main script for demo.
+"""
+
+from aitrika.engine.online_aitrika import OnlineAItrika
+from aitrika.llm.openai import OpenAILLM
+from aitrika.utils.text_parser import generate_documents
+from dotenv import load_dotenv
+import os
+
+if __name__ == "__main__":
+    load_dotenv()
+    pubmed_id = 23747889
+    engine = OnlineAItrika(pubmed_id=pubmed_id)
+
+    # Extract the content
+    abstract = engine.extract_abstract()
+
+    # Prepare the documents (you can use full-text if available)
+    documents = generate_documents(content=abstract)
+
+    # Set the LLM (default for OpenAI is gpt-4o-mini)
+    llm = OpenAILLM(
+        documents=documents,
+        api_key=os.getenv("OPENAI_API_KEY"),
+    )
+
+    # Query your document
+    query = "Is BRCA1 associated with breast cancer?"
+    print(llm.query(query=query))
+
+    # Extract paper results
+    results = engine.extract_results(llm=llm)
+    print(results)
+
+    # Extract number of participants
+    number_of_participants = engine.extract_number_of_participants(llm=llm)
+    print(number_of_participants)
+
+    # Extract outcomes
+    outcomes = engine.extract_outcomes(llm=llm)
+    print(outcomes)