Initial Commit

Src/api/fastapi_app.py

@@ -1,289 +1,3 @@
-# # from fastapi import FastAPI, Query
-# # from pydantic import BaseModel
-# # from datetime import datetime
-# # from ..services.db import init_db, seed_data, get_connection
-# # from ..services.patient_service import register_patient as save_patient
-# # from ..services.medicine_service import check_medicine_availability
-# # from ..rag.rag_pipeline import rag_query_multimodal
-# # from ..services.doctor_assignment import assign_doctor_with_gemma
-
-# # app = FastAPI()
-
-# # # ----------------------------
-# # #  Startup: Init DB & Seed
-# # # ----------------------------
-# # @app.on_event("startup")
-# # def startup_event():
-# #     init_db()
-# #     seed_data()
-
-# # # ----------------------------
-# # # 1. Chatbot Endpoint (RAG)
-# # # ----------------------------
-# # @app.get("/query")
-# # def query_bot(q: str = Query(..., description="Medical question")):
-# #     answer, references = rag_query_multimodal(q, k=5)
-# #     return {"answer": answer, "references": references}
-
-# # # ----------------------------
-# # # 2. Register Patient + Assign Doctor
-# # # ----------------------------
-# # class PatientData(BaseModel):
-# #     name: str
-# #     age: int
-# #     reason: str
-
-# # @app.post("/register_patient")
-# # def register_patient_api(data: PatientData):
-# #     # Assign doctor using Gemma
-# #     doctor, reasoning = assign_doctor_with_gemma(data.reason)
-
-# #     if not doctor:
-# #         return {"message": "No suitable doctor found. Please try again later."}
-
-# #     # Save patient and doctor_id to DB
-# #     patient_id = save_patient(data.name, data.age, data.reason, doctor["id"])
-
-# #     # Convert doctor row to dict
-# #     doctor_dict = {key: doctor[key] for key in doctor.keys()} if doctor else None
-
-# #     return {
-# #         "patient_id": patient_id,
-# #         "assigned_doctor": doctor_dict,
-# #         "reasoning": reasoning
-# #     }
-
-# # # ----------------------------
-# # # 3. Check Registration & Confirm Appointment
-# # # ----------------------------
-# # class AppointmentData(BaseModel):
-# #     name: str  # Patient name
-
-# # @app.post("/check_registration_status")
-# # def check_registration_status(data: AppointmentData):
-# #     conn = get_connection()
-# #     cur = conn.cursor()
-
-# #     # Get patient and assigned doctor
-# #     cur.execute("""
-# #         SELECT p.id AS patient_id, p.name AS patient_name, d.id AS doctor_id, 
-# #                d.name AS doctor_name, d.specialization, d.available
-# #         FROM patients p
-# #         LEFT JOIN doctors d ON p.doctor_id = d.id
-# #         WHERE p.name = ?
-# #     """, (data.name,))
-# #     record = cur.fetchone()
-
-# #     if not record:
-# #         conn.close()
-# #         return {"message": "Patient not registered. Please register first."}
-
-# #     if not record["doctor_id"]:
-# #         conn.close()
-# #         return {"message": "No doctor assigned yet. Please register again."}
-
-# #     # Mark doctor as unavailable if available
-# #     if record["available"] == 1:
-# #         cur.execute(
-# #             "UPDATE doctors SET available=0, last_booked_at=? WHERE id=?",
-# #             (datetime.now().isoformat(), record["doctor_id"])
-# #         )
-# #         conn.commit()
-
-# #     conn.close()
-
-# #     return {
-# #         "message": f"Appointment confirmed with {record['doctor_name']} ({record['specialization']}) for {record['patient_name']}."
-# #     }
-
-# # # ----------------------------
-# # # 4. Medicine Availability
-# # # ----------------------------
-# # @app.get("/medicine_availability")
-# # def medicine_availability_api(name: str = Query(..., description="Medicine name")):
-# #     result = check_medicine_availability(name)
-# #     return {"message": result}
-
-
-
-
-
-
-
-
-
-
-# # Src/api/fastapi_app.py
-
-# from fastapi import FastAPI, Query, Header, HTTPException
-
-# from pydantic import BaseModel
-# from datetime import datetime
-
-# # DB and services
-# from ..services.db import init_db, seed_data, get_connection
-# from ..services.patient_service import register_patient as save_patient, get_patient_full_case
-# from ..services.medicine_service import check_medicine_availability
-# from ..services.doctor_assignment import assign_doctor_with_gemma
-# from ..services.summarizer import summarize_patient_case
-# # RAG pipeline
-# from ..rag.rag_pipeline import rag_query_multimodal
-
-# # Agent system
-# from ..agent.orchestrator import orchestrate_query
-# from ..agent.agent_executor import get_agent_executor
-
-# # Initialize FastAPI app
-# app = FastAPI()
-
-# # ----------------------------
-# #  Startup: Init DB & Seed
-# # ----------------------------
-# @app.on_event("startup")
-# def startup_event():
-#     init_db()
-#     seed_data()
-
-# # ----------------------------
-# # 1. Medical RAG Chatbot
-# # ----------------------------
-# @app.get("/query")
-# def query_bot(q: str = Query(..., description="Medical question"), authorization: str = Header(...)):
-#     """
-#     Query the medical knowledge base (PDF + captions) using RAG.
-#     """
-#     hf_token = authorization.replace("Bearer ", "")
-#     answer, references = rag_query_multimodal(q, k=10, hf_token=hf_token)
-#     return {"answer": answer, "references": references}
-
-# # ----------------------------
-# # 2. Register Patient + Assign Doctor
-# # ----------------------------
-# class PatientData(BaseModel):
-#     name: str
-#     age: int
-#     reason: str
-
-# @app.post("/register_patient")
-# def register_patient_api(data: PatientData, authorization: str = Header(...)):
-#     """
-#     Register patient and assign doctor using Gemma LLM.
-#     """
-#     # Assign doctor via LLM
-#     hf_token = authorization.replace("Bearer ", "")
-#     doctor, reasoning = assign_doctor_with_gemma(data.reason, hf_token=hf_token)
-
-#     if not doctor:
-#         return {"message": "No suitable doctor found. Please try again later."}
-
-#     # Save patient record with doctor_id
-#     patient_record = save_patient({"name": data.name, "age": data.age, "reason": data.reason}, doctor["id"])
-
-#     return {
-#         "patient_id": patient_record["id"],
-#         "assigned_doctor": doctor,
-#         "reasoning": reasoning
-#     }
-
-# # ----------------------------
-# # 3. Check Registration & Confirm Appointment
-# # ----------------------------
-# class AppointmentData(BaseModel):
-#     name: str  # Patient name
-
-# @app.post("/check_registration_status")
-# def check_registration_status(data: AppointmentData):
-#     """
-#     Confirm appointment for registered patient with already assigned doctor.
-#     """
-#     conn = get_connection()
-#     cur = conn.cursor()
-
-#     # Get patient + doctor details
-#     cur.execute("""
-#         SELECT p.id AS patient_id, p.name AS patient_name, d.id AS doctor_id, 
-#                d.name AS doctor_name, d.specialization, d.available
-#         FROM patients p
-#         LEFT JOIN doctors d ON p.doctor_id = d.id
-#         WHERE p.name = ?
-#     """, (data.name,))
-#     record = cur.fetchone()
-
-#     if not record:
-#         conn.close()
-#         return {"message": "Patient not registered. Please register first."}
-
-#     if not record["doctor_id"]:
-#         conn.close()
-#         return {"message": "No doctor assigned yet. Please register again."}
-
-#     # Mark doctor unavailable if available
-#     if record["available"] == 1:
-#         cur.execute(
-#             "UPDATE doctors SET available=0, last_booked_at=? WHERE id=?",
-#             (datetime.now().isoformat(), record["doctor_id"])
-#         )
-#         conn.commit()
-
-#     conn.close()
-
-#     return {
-#         "message": f"Appointment confirmed with {record['doctor_name']} ({record['specialization']}) for {record['patient_name']}."
-#     }
-
-# # ----------------------------
-# # 4. Medicine Availability
-# # ----------------------------
-# @app.get("/medicine_availability")
-# def medicine_availability_api(name: str = Query(..., description="Medicine name")):
-#     """
-#     Check if a medicine is available in the pharmacy stock.
-#     """
-#     result = check_medicine_availability(name)
-#     return {"message": result}
-
-# # ----------------------------
-# # 5. Summarize Patient case
-# # ----------------------------
-# @app.get("/summarize_case/{patient_id}")
-# def summarize_case_api(patient_id: int):
-#     patient_data = get_patient_full_case(patient_id)
-#     if not patient_data:
-#         return {"message": "Patient not found"}
-#     summary = summarize_patient_case(patient_data)
-#     return {"summary": summary}
-
-
-# # ----------------------------
-# # 6. Agent Endpoint (Full LangChain Agent)
-# # ----------------------------
-# @app.get("/agent_query")
-# def agent_query(q: str = Query(..., description="General hospital or medical query")):
-#     """
-#     Use full LangChain agent (tools + RAG) to process queries.
-#     """
-#     agent = get_agent_executor()
-#     response = agent.run(q)
-#     return {"response": response}
-
-# # ----------------------------
-# # 6. Simple Orchestrator Endpoint
-# # ----------------------------
-# @app.get("/orchestrator_query")
-# def orchestrator_query(q: str = Query(..., description="Simple classified query")):
-#     """
-#     Use lightweight orchestrator to classify and execute tools or RAG.
-#     """
-#     result, references = orchestrate_query(q)
-#     return {"result": result, "references": references}
-
-
-
-
-
-
-
-
 from fastapi import FastAPI, Query, Header, HTTPException
 from pydantic import BaseModel
 from datetime import datetime
@@ -312,6 +26,10 @@ def startup_event():
     init_db()
     seed_data()
 
+@app.get("/")
+def read_root():
+    return {"Hello": "World!"}
+
 # ----------------------------
 # 1. Medical RAG Chatbot
 # ----------------------------