Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from duckduckgo_search import DDGS
-from typing import List, Dict
+from typing import List, Dict, Tuple
 from huggingface_hub import InferenceClient
 import os
 
@@ -51,7 +51,7 @@ def rephrase_query(previous_response: str, new_query: str) -> str:
     response = client.text_generation(prompt, max_new_tokens=100, temperature=0.7)
     return response
 
-def web_search_and_summarize(input_text: str, chat_history: List[str]) -> str:
+def web_search_and_summarize(input_text: str, chat_history: List[Tuple[str, str]]) -> Tuple[str, List[Tuple[str, str]]]:
     if chat_history:
         previous_response = chat_history[-1][1]
         rephrased_query = rephrase_query(previous_response, input_text)
@@ -60,20 +60,22 @@ def web_search_and_summarize(input_text: str, chat_history: List[str]) -> str:
 
     search_results = get_web_search_results(rephrased_query)
     if "error" in search_results[0]:
-        return search_results[0]["error"]
+        summary = search_results[0]["error"]
+    else:
+        summary = summarize_results(rephrased_query, search_results)
     
-    summary = summarize_results(rephrased_query, search_results)
-    return summary
+    chat_history.append((input_text, summary))
+    return summary, chat_history
 
 iface = gr.Interface(
     fn=web_search_and_summarize,
     inputs=[
         gr.Textbox(lines=2, placeholder="Enter your query here..."),
-        "state"
+        gr.State([])
     ],
     outputs=[
         gr.Textbox(label="Response"),
-        "state"
+        gr.State()
     ],
     title="DuckDuckGo Web Search and Summary",
     description="Enter a query to get a summary based on web search results using DuckDuckGo.",