File size: 2,386 Bytes
cbc783a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
from fastapi import FastAPI, Query, Request
from fastapi.responses import HTMLResponse
from fastapi.templating import Jinja2Templates
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from utils.vector_search import get_similar_symptoms
from utils.sql_queries import get_diseases_by_symptoms, get_related_symptoms
import sqlite3
app = FastAPI()
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # Allow all origins
allow_credentials=True,
allow_methods=["*"], # Allow all methods
allow_headers=["*"], # Allow all headers
)
templates = Jinja2Templates(directory="templates")
@app.get("/", response_class=HTMLResponse)
def index(request: Request):
return templates.TemplateResponse("index.html", {"request": request})
# Helper function to check if a symptom exists in the database
def is_valid_symptom(symptom):
conn = sqlite3.connect("disease.db")
cur = conn.cursor()
cur.execute("SELECT COUNT(*) FROM symptom WHERE name = ?", (symptom,))
count = cur.fetchone()[0]
conn.close()
return count > 0
@app.get("/search")
def search(symptom: str, selected: list[str] = Query([])):
# Get similar symptoms from vector search
similar = get_similar_symptoms(symptom)
# Check if the exact symptom exists in our database
if is_valid_symptom(symptom):
# Use the exact symptom if it exists
search_symptoms = [symptom] + selected
else:
# Otherwise, try to use the first similar symptom that exists in our database
valid_symptom_found = False
for sim in similar:
if is_valid_symptom(sim):
search_symptoms = [sim] + selected
valid_symptom_found = True
break
# If no valid symptom found, just use all similar symptoms
if not valid_symptom_found:
search_symptoms = similar + selected
# Get diseases that have ALL the selected search symptoms
diseases = get_diseases_by_symptoms(search_symptoms)
# Get related symptoms from the matching diseases, excluding already selected symptoms
suggestions = get_related_symptoms(diseases, exclude=search_symptoms)
return {
"matching_diseases": diseases,
"related_symptoms": suggestions,
"semantic_matches": similar
}
|