Update app.py

app.py

@@ -12,6 +12,18 @@ MODELS = [
     "mistralai/Mistral-Nemo-Instruct-2407"
 ]
 
+class ConversationManager:
+    def __init__(self):
+        self.history = []
+        self.current_context = None
+
+    def add_interaction(self, query, response):
+        self.history.append((query, response))
+        self.current_context = f"Previous query: {query}\nPrevious response summary: {response[:200]}..."
+
+    def get_context(self):
+        return self.current_context
+
 def get_web_search_results(query: str, max_results: int = 10) -> List[Dict[str, str]]:
     try:
         results = list(DDGS().text(query, max_results=max_results))
@@ -22,16 +34,34 @@ def get_web_search_results(query: str, max_results: int = 10) -> List[Dict[str,
         print(f"An error occurred during web search: {str(e)}")
         return [{"error": f"An error occurred during web search: {str(e)}"}]
 
-def summarize_results(query: str, search_results: List[Dict[str, str]], model: str) -> 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:
+
+        Context: {context}
+        
+        New query: {original_query}
+        
+        Rephrased query:"""
+        
+        rephrased_query = DDGS().chat(prompt, model="llama-3-70b")
+        return rephrased_query.strip()
+    return original_query
+
+def summarize_results(query: str, search_results: List[Dict[str, str]], model: str, conversation_manager: ConversationManager) -> str:
     try:
-        context = "\n\n".join([f"Title: {result['title']}\nContent: {result['body']}" for result in search_results])
+        context = conversation_manager.get_context()
+        search_context = "\n\n".join([f"Title: {result['title']}\nContent: {result['body']}" for result in search_results])
         
-        prompt = f"""Based on the following web search results about '{query}', please create a comprehensive news article. 
+        prompt = f"""Based on the following web search results about '{query}' and considering the context: {context}, 
+        please create a comprehensive news article. 
         Include key facts, relevant statistics, and expert opinions if available. 
         Ensure the article is well-structured with an introduction, main body, and conclusion. 
+        Address the query in the context of the ongoing conversation if applicable.
         Cite sources directly within the generated text and not at the end of the generated text, integrating URLs where appropriate to support the information provided:
         
-        {context}
+        {search_context}
         
         Article:"""
 
@@ -40,21 +70,30 @@ def summarize_results(query: str, search_results: List[Dict[str, str]], model: s
     except Exception as e:
         return f"An error occurred during summarization: {str(e)}"
 
+conversation_manager = ConversationManager()
+
 def respond(message, chat_history, model, temperature, num_api_calls):
     final_summary = ""
+    rephrased_query = rephrase_query(message, conversation_manager)
+    
     for _ in range(num_api_calls):
-        search_results = get_web_search_results(message)
+        search_results = get_web_search_results(rephrased_query)
         
         if not search_results:
-            final_summary += f"No search results found for the query: {message}\n\n"
+            final_summary += f"No search results found for the query: {rephrased_query}\n\n"
         elif "error" in search_results[0]:
             final_summary += search_results[0]["error"] + "\n\n"
         else:
-            summary = summarize_results(message, search_results, model)
+            summary = summarize_results(rephrased_query, search_results, model, conversation_manager)
             final_summary += summary + "\n\n"
 
-    return final_summary if final_summary else "Unable to generate a response. Please try a different query."
+    if final_summary:
+        conversation_manager.add_interaction(message, final_summary)
+        return final_summary
+    else:
+        return "Unable to generate a response. Please try a different query."
 
+# The rest of your code (CSS, theme, and Gradio interface setup) remains the same
 css = """
 Your custom CSS here
 """
@@ -107,4 +146,4 @@ demo = gr.ChatInterface(
     )
 )
 
-demo.launch()
+demo.launch()