Update app.py

app.py

@@ -588,17 +588,18 @@ async def resources_endpoint(profile: MedicalProfile):
                 title = get_page_title(original_url) or "Unknown Title"
                 resources.append({"file_name": file_name, "title": title, "url": original_url})
             else:
-                resources.append({"file_name": file_name, "title": "Unknown", "url": None})
+                resources.append({"file_name": file_name, "title": "Unknown", "url": None})        
         document_texts = retrieve_document_texts(document_ids, folder_path)
         if not document_texts:
-            raise ValueError("Failed to retrieve document texts.")
+            raise ValueError("Failed to retrieve document texts.")        
         cross_encoder = models['cross_encoder']
         scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
-        scores = [float(score) for score in scores]  
+        scores = [float(score) for score in scores]       
         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)
-        return {"resources": resources, "success": True}
+            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]       
+        return output
     except ValueError as ve:
         raise HTTPException(status_code=400, detail=str(ve))
     except Exception as e:
@@ -630,10 +631,16 @@ async def recipes_endpoint(profile: MedicalProfile):
         metadata_path = 'recipes_metadata.xlsx'
         metadata = retrieve_metadata(document_indices, metadata_path=metadata_path)
         print(f"Retrieved Metadata: {metadata}")        
-        response = {           
-            "recipes": metadata
-        }
-        return response       
+        recipes = []
+        for item in metadata:
+            recipes.append({
+                "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: