Spaces:
Sleeping
Sleeping
Commit
·
28f8ba5
1
Parent(s):
de5f21f
Update app.py
Browse files
app.py
CHANGED
@@ -633,31 +633,71 @@ async def resources_endpoint(profile: MedicalProfile):
|
|
633 |
@app.post("/api/recipes")
|
634 |
async def recipes_endpoint(profile: MedicalProfile):
|
635 |
try:
|
636 |
-
|
637 |
-
|
638 |
-
|
639 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
640 |
folder_path = 'downloaded_articles/downloaded_articles'
|
641 |
initial_results = query_embeddings(query_embedding, embeddings_data, n_results=5)
|
|
|
|
|
|
|
|
|
642 |
document_ids = [doc_id for doc_id, _ in initial_results]
|
|
|
|
|
643 |
document_texts = retrieve_document_texts(document_ids, folder_path)
|
|
|
|
|
|
|
|
|
644 |
cross_encoder = models['cross_encoder']
|
645 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
|
|
|
|
|
|
646 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
647 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
648 |
recipes = []
|
649 |
-
for
|
650 |
-
|
651 |
-
|
652 |
-
|
653 |
-
|
654 |
-
|
655 |
-
|
656 |
-
|
657 |
-
|
|
|
|
|
|
|
|
|
|
|
658 |
return {"recipes": recipes[:5], "success": True}
|
|
|
|
|
|
|
|
|
659 |
except Exception as e:
|
660 |
-
|
|
|
|
|
661 |
|
662 |
if not init_success:
|
663 |
print("Warning: Application initialized with partial functionality")
|
|
|
633 |
@app.post("/api/recipes")
|
634 |
async def recipes_endpoint(profile: MedicalProfile):
|
635 |
try:
|
636 |
+
# Build the query text for recipes
|
637 |
+
recipe_query = (
|
638 |
+
f"Recipes and meals suitable for someone with: "
|
639 |
+
f"{profile.conditions} and experiencing {profile.daily_symptoms}"
|
640 |
+
)
|
641 |
+
query_text = recipe_query
|
642 |
+
|
643 |
+
# Generate the query embedding
|
644 |
+
query_embedding = embed_query_text(query_text)
|
645 |
+
if query_embedding is None:
|
646 |
+
raise ValueError("Failed to generate query embedding.")
|
647 |
+
|
648 |
+
# Load embeddings and retrieve initial results
|
649 |
+
embeddings_data = load_embeddings()
|
650 |
folder_path = 'downloaded_articles/downloaded_articles'
|
651 |
initial_results = query_embeddings(query_embedding, embeddings_data, n_results=5)
|
652 |
+
if not initial_results:
|
653 |
+
raise ValueError("No relevant recipes found.")
|
654 |
+
|
655 |
+
# Extract document IDs
|
656 |
document_ids = [doc_id for doc_id, _ in initial_results]
|
657 |
+
|
658 |
+
# Retrieve document texts
|
659 |
document_texts = retrieve_document_texts(document_ids, folder_path)
|
660 |
+
if not document_texts:
|
661 |
+
raise ValueError("Failed to retrieve document texts.")
|
662 |
+
|
663 |
+
# Perform re-ranking with cross-encoder
|
664 |
cross_encoder = models['cross_encoder']
|
665 |
scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
|
666 |
+
scores = [float(score) for score in scores] # Convert scores to native floats
|
667 |
+
|
668 |
+
# Combine document data
|
669 |
scored_documents = list(zip(scores, document_ids, document_texts))
|
670 |
+
scored_documents.sort(key=lambda x: x[0], reverse=True) # Sort by score
|
671 |
+
|
672 |
+
# Load recipe metadata from DataFrame
|
673 |
+
file_path = 'finalcleaned_excel_file.xlsx'
|
674 |
+
df = pd.read_excel(file_path)
|
675 |
+
|
676 |
+
# Prepare the final recipes list
|
677 |
recipes = []
|
678 |
+
for score, doc_id, text in scored_documents:
|
679 |
+
# Retrieve metadata for the document
|
680 |
+
doc_info = df[df['Unnamed: 0'] == doc_id]
|
681 |
+
if not doc_info.empty:
|
682 |
+
title = doc_info.iloc[0]['title'] if 'title' in doc_info.columns else "Unknown Title"
|
683 |
+
if 'recipe' in text.lower() or 'meal' in text.lower():
|
684 |
+
recipes.append({
|
685 |
+
"id": doc_id,
|
686 |
+
"title": title,
|
687 |
+
"content_preview": text[:200], # First 200 characters of text
|
688 |
+
"score": score,
|
689 |
+
})
|
690 |
+
|
691 |
+
# Limit the response to top 5 recipes
|
692 |
return {"recipes": recipes[:5], "success": True}
|
693 |
+
|
694 |
+
except ValueError as ve:
|
695 |
+
# Handle expected errors
|
696 |
+
raise HTTPException(status_code=400, detail=str(ve))
|
697 |
except Exception as e:
|
698 |
+
# Handle unexpected errors
|
699 |
+
print(f"Unexpected error: {e}")
|
700 |
+
raise HTTPException(status_code=500, detail="An unexpected error occurred.")
|
701 |
|
702 |
if not init_success:
|
703 |
print("Warning: Application initialized with partial functionality")
|