Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 from duckduckgo_search import DDGS
 from datetime import datetime
 
-# Initialize a lightweight text generation model on CPU (moves to GPU when decorated)
+# Initialize a lightweight text generation model on CPU
 generator = pipeline("text-generation", model="distilgpt2", device=-1)  # -1 ensures CPU by default
 
 # Web search function (CPU-based)
@@ -57,7 +57,7 @@ def process_deep_research(query: str, history: list):
     """Handle the deep research process."""
     if not history:
         history = []
-
+    
     # Fetch web results (CPU)
     web_results = get_web_results(query)
     sources_html = format_sources(web_results)
@@ -65,9 +65,11 @@ def process_deep_research(query: str, history: list):
     # Generate answer (GPU via @spaces.GPU)
     prompt = format_prompt(query, web_results)
     answer = generate_answer(prompt)
-    final_history = history + [[query, answer]]
-
-    return answer, sources_html, final_history
+    
+    # Convert tuple history to messages format (role/content)
+    new_history = history + [{"role": "user", "content": query}, {"role": "assistant", "content": answer}]
+    
+    return answer, sources_html, new_history
 
 # Custom CSS for a cool, lightweight UI
 css = """
@@ -163,26 +165,27 @@ with gr.Blocks(title="Deep Research Engine - ZeroGPU", css=css) as demo:
         with gr.Column():
             sources_output = gr.HTML(label="Sources", elem_classes="sources-list")
 
-    # Chat history
+    # Chat history (using messages format)
     with gr.Row():
-        history_display = gr.Chatbot(label="History", elem_classes="history-box")
+        history_display = gr.Chatbot(label="History", elem_classes="history-box", type="messages")
 
     # Event handling
     def handle_search(query, history):
         answer, sources, new_history = process_deep_research(query, history)
         return answer, sources, new_history
 
+    # Remove _js parameter from click event
     search_btn.click(
         fn=handle_search,
         inputs=[search_input, history_state],
-        outputs=[answer_output, sources_output, history_display],
-        _js="() => [document.querySelector('.search-box input').value, null]"
+        outputs=[answer_output, sources_output, history_display]
     ).then(
         fn=lambda x: x,
         inputs=[history_display],
         outputs=[history_state]
     )
 
+    # Remove _js parameter from submit event
     search_input.submit(
         fn=handle_search,
         inputs=[search_input, history_state],