thechaiexperiment commited on
Commit
c15ec08
·
1 Parent(s): b51f89f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -14
app.py CHANGED
@@ -593,20 +593,7 @@ async def resources_endpoint(profile: MedicalProfile):
593
  scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
594
  scored_documents = list(zip(scores, document_ids, document_texts))
595
  scored_documents.sort(key=lambda x: x[0], reverse=True)
596
- # Build resources response
597
- resources = []
598
- for score, doc_id, text in scored_documents[:10]:
599
- doc_info = data['df'][data['df']['id'] == doc_id]
600
- if doc_info.empty:
601
- continue # Skip missing documents
602
- doc_info = doc_info.iloc[0] # Safe to access the first row now
603
- resources.append({
604
- "id": doc_id,
605
- "title": doc_info['title'],
606
- "content": text[:200],
607
- "score": float(score)
608
- })
609
- return {"resources": resources, "success": True}
610
 
611
 
612
  @app.post("/api/recipes")
 
593
  scores = cross_encoder.predict([(query_text, doc) for doc in document_texts])
594
  scored_documents = list(zip(scores, document_ids, document_texts))
595
  scored_documents.sort(key=lambda x: x[0], reverse=True)
596
+ return {"resources": scored_documents, "success": True}
 
 
 
 
 
 
 
 
 
 
 
 
 
597
 
598
 
599
  @app.post("/api/recipes")