Update app.py
Browse files
app.py
CHANGED
@@ -48,20 +48,19 @@ def predict(message, history, selected_agent):
|
|
48 |
# Define the Gradio interface using Blocks for layout
|
49 |
def app_interface():
|
50 |
with gr.Blocks() as demo:
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
)
|
65 |
return demo
|
66 |
|
67 |
# Launch the app
|
|
|
48 |
# Define the Gradio interface using Blocks for layout
|
49 |
def app_interface():
|
50 |
with gr.Blocks() as demo:
|
51 |
+
dropdown = gr.Dropdown(
|
52 |
+
choices=list(extractor_agents.keys()),
|
53 |
+
value="Solution Specifier A",
|
54 |
+
label="Choose Extractor Agent",
|
55 |
+
interactive=True,
|
56 |
+
)
|
57 |
+
|
58 |
+
chat = gr.ChatInterface(
|
59 |
+
fn=predict,
|
60 |
+
additional_inputs=[dropdown],
|
61 |
+
title="Solution Specifier Chat",
|
62 |
+
description="Test with different solution specifiers"
|
63 |
+
)
|
|
|
64 |
return demo
|
65 |
|
66 |
# Launch the app
|