Update app.py

app.py

@@ -114,16 +114,15 @@ def respond(message, chat_history, model, temperature, num_api_calls):
 
     if 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."
-
+        
         # Extract image URL from the summary
-    image_url = None
-    if "Image found:" in final_summary:
-        image_url = final_summary.split("URL: ")[1].split("\n")[0]
-    
-    return final_summary, image_url
+        image_url = None
+        if "Image found:" in final_summary:
+            image_url = final_summary.split("URL: ")[1].split("\n")[0]
+        
+        return final_summary, image_url
+    else:
+        return "Unable to generate a response. Please try a different query.", None
 
 # The rest of your code (CSS, theme, and Gradio interface setup) remains the same
 css = """
@@ -151,37 +150,39 @@ theme = gr.themes.Soft(
     code_background_fill_dark="#140b0b"
 )
 
-demo = gr.Interface(
-    fn=respond,
-    inputs=[
-        gr.Textbox(placeholder=custom_placeholder, label="Message"),
-        gr.State([]),  # For chat history
-        gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[2]),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature"),
-        gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls")
-    ],
-    outputs=[
-        gr.Textbox(label="Response"),
-        gr.Image(label="Related Image")
-    ],
-    title="AI-powered Web Search and PDF Chat Assistant",
-    description="This AI-powered Web Search and PDF Chat Assistant combines real-time web search capabilities with advanced language processing.",
-    theme=theme,
-    css=css,
-    examples=[
-        ["What is AI"],
-        ["Any recent news on US Banks"],
-        ["Who is Donald Trump"]
-    ],
-    cache_examples=False,
-    analytics_enabled=False,
-    textbox=gr.Textbox(placeholder=custom_placeholder, container=False, scale=7),
-    chatbot=gr.Chatbot(
-        show_copy_button=True,
-        likeable=True,
-        layout="bubble",
-        height=400,
+with gr.Blocks(theme=theme, css=css) as demo:
+    gr.Markdown("# AI-powered Web Search and PDF Chat Assistant")
+    gr.Markdown("This AI-powered Web Search and PDF Chat Assistant combines real-time web search capabilities with advanced language processing.")
+
+    with gr.Row():
+        with gr.Column(scale=7):
+            chatbot = gr.Chatbot(
+                show_copy_button=True,
+                likeable=True,
+                layout="bubble",
+                height=400,
+            )
+            msg = gr.Textbox(placeholder=custom_placeholder, container=False, scale=7)
+            clear = gr.Button("Clear")
+
+        with gr.Column(scale=3):
+            model = gr.Dropdown(choices=MODELS, label="Select Model", value=MODELS[2])
+            temperature = gr.Slider(minimum=0.1, maximum=1.0, value=0.2, step=0.1, label="Temperature")
+            num_api_calls = gr.Slider(minimum=1, maximum=5, value=1, step=1, label="Number of API Calls")
+            image_output = gr.Image(label="Related Image")
+
+    def user(user_message, history):
+        return "", history + [[user_message, None]]
+
+    def bot(history, model, temperature, num_api_calls):
+        user_message = history[-1][0]
+        bot_response, image_url = respond(user_message, history, model, temperature, num_api_calls)
+        history[-1][1] = bot_response
+        return history, image_url
+
+    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        bot, [chatbot, model, temperature, num_api_calls], [chatbot, image_output]
     )
-)
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 demo.launch()