Update app.py

app.py

@@ -45,10 +45,8 @@ def handle_message(message: str, mode: str):
     else:
         result_text = "Select a valid mode."
     
-    result_combined = f"Image URL: {result_image}\n\n{result_text}" if result_image else result_text
+    return result_text, result_image
     
-    return result_combined
-
 def chat_message(message: str):
     global messages
     
@@ -114,11 +112,17 @@ def chart_generator(message: str):
         return f"IT-Assistant: {response.content}", chart_url
     else:
         return f"Can't generate this image. Please provide valid chart details.", None
+
+def custom_display_text_and_image(result_text, result_image):
+    if result_image:
+        display(HTML(f"<p>{result_text}</p><img src='{result_image}'/>"))
+    else:
+        print(result_text)
         
 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"),
+    outputs=custom_display_text_and_image,
     title="IT Assistant")
 
 demo.launch()