Update app.py

app.py

@@ -45,7 +45,9 @@ def handle_message(message: str, mode: str):
     else:
         result_text = "Select a valid mode."
     
-    return result_text, result_image
+    result_combined = f"Image URL: {result_image}\n\n{result_text}" if result_image else result_text
+    
+    return result_combined
 
 def chat_message(message: str):
     global messages
@@ -116,7 +118,7 @@ def chart_generator(message: str):
 demo = gr.Interface(
     fn=handle_message,
     inputs=["text", gr.Radio(["Chat-Message", "Web-Search", "Chart-Generator"], label="mode", info="Choose a mode and enter your message, then click submit to interact.")],
-    outputs=[gr.Textbox(label="Response"), gr.Image(type="filepath")],
+    outputs=gr.Textbox(label="Response"),
     title="IT Assistant")
 
 demo.launch()