Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 from duckduckgo_search import DDGS
 from typing import List, Dict
 import os
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
 
 # Environment variables and configurations
 huggingface_token = os.environ.get("HUGGINGFACE_TOKEN")
@@ -37,7 +41,7 @@ def get_web_search_results(query: str, max_results: int = 10) -> List[Dict[str,
 def rephrase_query(original_query: str, conversation_manager: ConversationManager) -> str:
     context = conversation_manager.get_context()
     if context:
-        prompt = f"""Given the following context and a new query, rephrase the query to make it more specific and contextual:
+        prompt = f"""Given the following context and a new query, rephrase the query to make it more specific and contextual. Provide ONLY the rephrased query without any additional explanation:
 
         Context: {context}
         
@@ -45,8 +49,10 @@ def rephrase_query(original_query: str, conversation_manager: ConversationManage
         
         Rephrased query:"""
         
-        rephrased_query = DDGS().chat(prompt, model="llama-3-70b")
-        return rephrased_query.strip()
+        response = DDGS().chat(prompt, model="llama-3-70b")
+        # Extract only the rephrased query, removing any explanations
+        rephrased_query = response.split('\n')[0].strip()
+        return rephrased_query
     return original_query
 
 def summarize_results(query: str, search_results: List[Dict[str, str]], model: str, conversation_manager: ConversationManager) -> str:
@@ -74,8 +80,12 @@ conversation_manager = ConversationManager()
 
 def respond(message, chat_history, model, temperature, num_api_calls):
     final_summary = ""
+    original_query = message
     rephrased_query = rephrase_query(message, conversation_manager)
     
+    logging.info(f"Original query: {original_query}")
+    logging.info(f"Rephrased query: {rephrased_query}")
+    
     for _ in range(num_api_calls):
         search_results = get_web_search_results(rephrased_query)
         
@@ -88,7 +98,7 @@ def respond(message, chat_history, model, temperature, num_api_calls):
             final_summary += summary + "\n\n"
 
     if final_summary:
-        conversation_manager.add_interaction(message, final_summary)
+        conversation_manager.add_interaction(original_query, final_summary)
         return final_summary
     else:
         return "Unable to generate a response. Please try a different query."