Expose api endpoints

api.py

@@ -0,0 +1,107 @@
+import os
+
+from fastapi import FastAPI, HTTPException
+from gradio_client import Client
+from enum import Enum
+from typing import List, Optional
+from pydantic import BaseModel, Field, constr, ConfigDict
+from fastapi.responses import FileResponse
+
+from docs import description, tags_metadata
+
+
+app = FastAPI(
+    title="ACRES RAG API",
+    description=description,
+    openapi_tags=tags_metadata,
+)
+client = Client("http://localhost:7860/")
+
+class StudyVariables(str, Enum):
+    ebola_virus = "Ebola Virus"
+    vaccine_coverage = "Vaccine coverage"
+    genexpert = "GeneXpert"
+
+
+class PromptType(str, Enum):
+    default = "Default"
+    highlight = "Highlight"
+    evidence_based = "Evidence-based"
+
+class StudyVariableRequest(BaseModel):
+    study_variable: StudyVariables
+    prompt_type: PromptType
+    text: constr(min_length=1, strip_whitespace=True)  # type: ignore
+
+    model_config = ConfigDict(from_attributes=True)
+
+class DownloadCSV(BaseModel):
+    text: constr(min_length=1, strip_whitespace=True)  # type: ignore
+
+    model_config = ConfigDict(from_attributes=True)
+
+
+class Study(BaseModel):
+    study_name: constr(min_length=1, strip_whitespace=True)  # type: ignore
+
+    model_config = ConfigDict(from_attributes=True)
+
+
+class ZoteroCredentials(BaseModel):
+    library_id: constr(min_length=1, strip_whitespace=True)  # type: ignore
+    api_access_key: constr(min_length=1, strip_whitespace=True)  # type: ignore
+
+    model_config = ConfigDict(from_attributes=True)
+
+
+@app.post("/process_zotero_library_items", tags=["zotero"])
+def process_zotero_library_items(zotero_credentials: ZoteroCredentials):
+    result = client.predict(
+            zotero_library_id=zotero_credentials.library_id,
+            zotero_api_access_key=zotero_credentials.api_access_key,
+            api_name="/process_zotero_library_items"
+    )
+    return {"result":result}
+
+
+
+@app.post("/get_study_info", tags=["zotero"])
+def get_study_info(study: Study):
+    result = client.predict(
+            study_name=study.study_name,
+            api_name="/get_study_info"
+    )
+    # print(result)
+    return {"result":result}
+
+
+@app.post("/study_variables", tags=["zotero"])
+def process_study_variables(study_request: StudyVariableRequest,):
+    result = client.predict(
+            text=study_request.text,  # "study id, study title, study design, study summary",
+            study_name=study_request.study_variable,  # "Ebola Virus",
+            prompt_type=study_request.prompt_type,  #"Default",
+            api_name="/process_multi_input"
+    )
+    print(type(result))
+    return {"result":result[0]}
+
+
+@app.post("/download_csv", tags=["zotero"])
+def download_csv(download_request: DownloadCSV):
+    result = client.predict(
+            markdown_content=download_request.text,
+            api_name="/download_as_csv"
+    )
+    print(result)
+    
+    file_path = result
+    if not file_path or not os.path.exists(file_path):
+        raise HTTPException(status_code=404, detail="File not found")
+
+    # Use FileResponse to send the file to the client
+    return FileResponse(
+        file_path,
+        media_type="text/csv",  # Specify the correct MIME type for CSV
+        filename=os.path.basename(file_path)  # Provide a default filename for the download
+    )

docs.py

@@ -0,0 +1,13 @@
+description = """
+Welcome to the Acres AI RAG API documentation.
+
+### RAG Tasks
+- Use the `/process_zotero_library_items`: Process zotero library items with your zotero credentials.
+- Use the `/get_study_info`: Get number of documents in a zotero study.
+- Use the `/study_variables`: Get research summary from the study provided the study variables.
+- Use the `/download_csv`: Export the markdown text to a csv file.
+"""
+
+tags_metadata = [
+    {"name": "ACRES RAG", "description": "AI RAG Application"},
+]

requirements.txt

@@ -1,6 +1,7 @@
 chromadb==0.5.5
 fastapi==0.112.2
 gradio
+gradio_client
 llama-index
 llama-index-vector-stores-chroma
 nest-asyncio==1.6.0

study_files.json

@@ -4,5 +4,7 @@
     "GeneXpert": "data/gene_xpert_zotero_items.json",
     "Zotero Collection Pastan": "data/zotero-collection-pastan_zotero_items.json",
     "EBSCOhost": "data/ebscohost_zotero_items.json",
-    "ExportedRis_file_1_of_1 (1)": "data/exportedris-file-1-of-1-1_zotero_items.json"
+    "ExportedRis_file_1_of_1 (1)": "data/exportedris-file-1-of-1-1_zotero_items.json",
+    "wb_1813-9450-6689": "data/wb-1813-9450-6689_zotero_items.json",
+    "kayongo papers": "data/kayongo-papers_zotero_items.json"
 }