Update main.py

main.py

@@ -59,11 +59,6 @@ def get_trial_info(nct_id):
 async def retrieve_trial(request: QueryRequest):
     """Retrieve Clinical Trial based on text"""
     query_vector = generate_embedding(request.text)
-    
-    # Check if FAISS index has vectors
-    if index.ntotal == 0:
-        return {"error": "FAISS index is empty. No trials available."}
-
     distances, indices = index.search(query_vector, request.top_k)
 
     matched_trials = []
@@ -73,10 +68,9 @@ async def retrieve_trial(request: QueryRequest):
             trial_data = get_trial_info(nct_id)
 
             if trial_data:
-                # **Final Fix: Ensure JSON-safe float values**
-                if np.isnan(dist) or np.isinf(dist):  
-                    print(f"⚠️ Skipping invalid distance: {dist}")  
-                    trial_data["similarity"] = 0.0  # Default to 0 if invalid
+                if not np.isfinite(dist):  # FIX: Handle NaN/Inf
+                    print(f"⚠️ Skipping invalid distance: {dist}")
+                    trial_data["similarity"] = 0.0
                 else:
                     trial_data["similarity"] = float(round(100 / (1 + dist), 2))