Update app.py

app.py

@@ -3,7 +3,6 @@ import pickle
 import os
 import re
 import numpy as np
-from fastapi.responses import HTMLResponse
 import torchvision
 import nltk
 import torch
@@ -35,7 +34,6 @@ from safetensors.torch import safe_open
 nltk.download('punkt_tab')
 
 app = FastAPI()
-    
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -507,8 +505,6 @@ def translate_en_to_ar(text):
 async def root():
     return {"message": "Welcome to the FastAPI application! Use the /health endpoint to check health, and /api/query for processing queries."}
 
-
-
 @app.get("/health")
 async def health_check():
     """Health check endpoint"""
@@ -602,8 +598,7 @@ async def resources_endpoint(profile: MedicalProfile):
         for i, resource in enumerate(resources):
             resource["score"] = scores[i] if i < len(scores) else 0.0        
         resources.sort(key=lambda x: x["score"], reverse=True)       
-        output = [{"title": resource["title"], "url": resource["url"]} for resource in resources]  
-        print (output)
+        output = [{"title": resource["title"], "url": resource["url"]} for resource in resources]       
         return output
     except ValueError as ve:
         raise HTTPException(status_code=400, detail=str(ve))
@@ -632,18 +627,20 @@ async def recipes_endpoint(profile: MedicalProfile):
         print("Initial results (document indices and similarities):")
         print(initial_results)
         document_indices = [doc_id for doc_id, _ in initial_results]
-        print("Document indices:", document_indices)
+        print("Document indices:", document_indices)        
         metadata_path = 'recipes_metadata.xlsx'
         metadata = retrieve_metadata(document_indices, metadata_path=metadata_path)
         print(f"Retrieved Metadata: {metadata}")        
         recipes = []
-        for item in metadata.values():
+        for item in metadata:
             recipes.append({
-                "title": item["original_file_name"] if "original_file_name" in item else "Unknown Title",
-                "url": item["url"] if "url" in item else ""
-            })               
-        print(recipes)
-        return recipes
+                "title": item.get("original_file_name", "Unknown Title"),
+                "url": item.get("url", "")
+            })        
+        response = {
+            "recipes": recipes
+        }        
+        return response   
     except ValueError as ve:
         raise HTTPException(status_code=400, detail=str(ve))
     except Exception as e: