Update app.py

app.py

@@ -69,7 +69,7 @@ def rephrase_query(previous_response: str, new_query: str) -> str:
     response = client.text_generation(prompt, max_new_tokens=30, temperature=0.3)
     
     # Extract only the rephrased query from the response
-    rephrased_query = response.split(":")[-1].strip()
+    rephrased_query = response.split("\n")[-1].strip()
     return rephrased_query
 
 def respond(message, chat_history, model, temperature, num_api_calls):
@@ -82,6 +82,10 @@ def respond(message, chat_history, model, temperature, num_api_calls):
 
     final_summary = ""
     for _ in range(num_api_calls):
+        if not rephrased_query or not isinstance(rephrased_query, str):
+            print(f"Invalid rephrased query: {rephrased_query}")
+            break
+
         search_results = get_web_search_results(rephrased_query)
         
         # If no results or error, try rephrasing and searching again