Update app.py

app.py

@@ -46,7 +46,7 @@ def handle_message(message: str, mode: str):
         result_text = "Select a valid mode."
     
     return result_text, result_image
-    
+
 def chat_message(message: str):
     global messages
     
@@ -73,9 +73,7 @@ def web_search(message: str):
         
     augmented_prompt = f"""
     If the answer to the next query is not contained in the Web Search, say 'No Answer Is Available' and then just give guidance for the query.
-
     Query: {message}
-
     Web Search:
     {source_knowledge}
     """
@@ -112,17 +110,11 @@ 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=custom_display_text_and_image,
+    outputs=[gr.Textbox(label="Response"), gr.Image(label="Chart if generated", type="filepath")],
     title="IT Assistant")
 
 demo.launch()