Update app.py

app.py

@@ -267,9 +267,19 @@ def rank_search_results(titles, summaries, model):
     
     ranking_prompt += "Ranks:"
     
-    ranks_str = generate_chunked_response(model, ranking_prompt)
-    ranks = [float(rank.strip()) for rank in ranks_str.split(',')]
-    return ranks
+    try:
+        ranks_str = generate_chunked_response(model, ranking_prompt)
+        ranks = [float(rank.strip()) for rank in ranks_str.split(',') if rank.strip()]
+        
+        # Check if we have the correct number of ranks
+        if len(ranks) != len(titles):
+            raise ValueError("Number of ranks does not match number of titles")
+        
+        return ranks
+    except Exception as e:
+        print(f"Error in ranking: {str(e)}. Using fallback ranking method.")
+        # Fallback: assign ranks based on original order
+        return list(range(1, len(titles) + 1))
 
 def ask_question(question, temperature, top_p, repetition_penalty, web_search):
     global conversation_history
@@ -311,8 +321,12 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
         # Rank the results
         titles = [r["title"] for r in processed_results]
         summaries = [r["summary"] for r in processed_results]
-        ranks = rank_search_results(titles, summaries, model)
-        
+        try:
+            ranks = rank_search_results(titles, summaries, model)
+        except Exception as e:
+            print(f"Error in ranking results: {str(e)}. Using default ranking.")
+            ranks = list(range(1, len(processed_results) + 1))
+
         # Update Vector DB
         current_date = datetime.now().strftime("%Y-%m-%d")
         update_vector_db_with_search_results(processed_results, ranks, current_date)