Update app.py
Browse files
app.py
CHANGED
@@ -45,10 +45,8 @@ def handle_message(message: str, mode: str):
|
|
45 |
else:
|
46 |
result_text = "Select a valid mode."
|
47 |
|
48 |
-
|
49 |
|
50 |
-
return result_combined
|
51 |
-
|
52 |
def chat_message(message: str):
|
53 |
global messages
|
54 |
|
@@ -114,11 +112,17 @@ def chart_generator(message: str):
|
|
114 |
return f"IT-Assistant: {response.content}", chart_url
|
115 |
else:
|
116 |
return f"Can't generate this image. Please provide valid chart details.", None
|
|
|
|
|
|
|
|
|
|
|
|
|
117 |
|
118 |
demo = gr.Interface(
|
119 |
fn=handle_message,
|
120 |
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.")],
|
121 |
-
outputs=
|
122 |
title="IT Assistant")
|
123 |
|
124 |
demo.launch()
|
|
|
45 |
else:
|
46 |
result_text = "Select a valid mode."
|
47 |
|
48 |
+
return result_text, result_image
|
49 |
|
|
|
|
|
50 |
def chat_message(message: str):
|
51 |
global messages
|
52 |
|
|
|
112 |
return f"IT-Assistant: {response.content}", chart_url
|
113 |
else:
|
114 |
return f"Can't generate this image. Please provide valid chart details.", None
|
115 |
+
|
116 |
+
def custom_display_text_and_image(result_text, result_image):
|
117 |
+
if result_image:
|
118 |
+
display(HTML(f"<p>{result_text}</p><img src='{result_image}'/>"))
|
119 |
+
else:
|
120 |
+
print(result_text)
|
121 |
|
122 |
demo = gr.Interface(
|
123 |
fn=handle_message,
|
124 |
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.")],
|
125 |
+
outputs=custom_display_text_and_image,
|
126 |
title="IT Assistant")
|
127 |
|
128 |
demo.launch()
|