Spaces:
Sleeping
Sleeping
gradio re interface
Browse files- Dockerfile +0 -13
- interface.py +51 -28
Dockerfile
DELETED
@@ -1,13 +0,0 @@
|
|
1 |
-
FROM python:3.10.9
|
2 |
-
|
3 |
-
RUN useradd -m -u 1000 user
|
4 |
-
USER user
|
5 |
-
ENV PATH="/home/user/.local/bin:$PATH"
|
6 |
-
|
7 |
-
WORKDIR /app
|
8 |
-
|
9 |
-
COPY --chown=user ./requirements.txt requirements.txt
|
10 |
-
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
11 |
-
|
12 |
-
COPY --chown=user . /app
|
13 |
-
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
interface.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
|
4 |
-
API_BASE_URL = "http://
|
5 |
|
6 |
# Function: Get nurse response
|
7 |
def get_nurse_response(user_input, model_name, chat_history):
|
@@ -83,25 +83,59 @@ def create_gradio_interface():
|
|
83 |
)
|
84 |
|
85 |
# Main Input Section
|
86 |
-
with gr.
|
87 |
-
with gr.Column():
|
88 |
chat_box = gr.Chatbot(label="Chat with MALI Nurse", scale=1)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
89 |
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
)
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
|
|
|
|
|
|
101 |
)
|
102 |
-
|
103 |
-
|
104 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
105 |
|
106 |
# Bind Get Nurse Response button
|
107 |
send_button.click(
|
@@ -117,16 +151,7 @@ def create_gradio_interface():
|
|
117 |
chat_history_button = gr.Button("View Chat History")
|
118 |
ehr_details_button = gr.Button("View EHR Details")
|
119 |
with gr.Column():
|
120 |
-
|
121 |
-
label="Chat History",
|
122 |
-
interactive=False,
|
123 |
-
lines=6,
|
124 |
-
)
|
125 |
-
ehr_details_output = gr.Textbox(
|
126 |
-
label="EHR Details",
|
127 |
-
interactive=False,
|
128 |
-
lines=6,
|
129 |
-
)
|
130 |
ehr_prompt_output = gr.Textbox(
|
131 |
label="Outputs",
|
132 |
interactive=False,
|
@@ -151,14 +176,12 @@ def create_gradio_interface():
|
|
151 |
)
|
152 |
|
153 |
send_button.click(
|
154 |
-
fn=view_chat_history,
|
155 |
fn=view_ehr_details,
|
156 |
inputs=[gr.Textbox(value="details", visible=False)],
|
157 |
outputs=ehr_details_output
|
158 |
)
|
159 |
|
160 |
send_button.click(
|
161 |
-
fn=view_chat_history,
|
162 |
fn=view_ehr_details,
|
163 |
inputs=[gr.Textbox(value="prompt", visible=False)],
|
164 |
outputs=ehr_prompt_output
|
|
|
1 |
import gradio as gr
|
2 |
import requests
|
3 |
|
4 |
+
API_BASE_URL = "http://localhost:8000" # Update this with your actual API base URL if deployed elsewhere
|
5 |
|
6 |
# Function: Get nurse response
|
7 |
def get_nurse_response(user_input, model_name, chat_history):
|
|
|
83 |
)
|
84 |
|
85 |
# Main Input Section
|
86 |
+
with gr.Column(scale=2):
|
|
|
87 |
chat_box = gr.Chatbot(label="Chat with MALI Nurse", scale=1)
|
88 |
+
send_button = gr.Button("Send", variant="primary", size="lg", scale=1)
|
89 |
+
with gr.Row():
|
90 |
+
user_input = gr.Textbox(
|
91 |
+
label="Your Message",
|
92 |
+
placeholder="Type your question or message here...",
|
93 |
+
lines=2,
|
94 |
+
)
|
95 |
+
model_name = gr.Radio(
|
96 |
+
choices=["typhoon-v1.5x-70b-instruct", "openthaigpt", "llama-3.3-70b-versatile"],
|
97 |
+
value="typhoon-v1.5x-70b-instruct",
|
98 |
+
label="Model Selection",
|
99 |
+
)
|
100 |
+
|
101 |
+
with gr.Column(scale=1):
|
102 |
+
output_selector = gr.Dropdown(
|
103 |
+
choices=["Chat History", "EHR Details"],
|
104 |
+
value="Chat History",
|
105 |
+
label="Select Output to Display",
|
106 |
+
)
|
107 |
|
108 |
+
chat_history_output = gr.Textbox(
|
109 |
+
label="Chat History Output",
|
110 |
+
interactive=False,
|
111 |
+
lines=6,
|
112 |
+
scale=1,
|
113 |
+
visible=True, # Initially visible
|
114 |
)
|
115 |
+
|
116 |
+
ehr_details_output = gr.Textbox(
|
117 |
+
label="EHR Details Output",
|
118 |
+
interactive=False,
|
119 |
+
lines=6,
|
120 |
+
scale=1,
|
121 |
+
visible=False, # Initially hidden
|
122 |
)
|
123 |
+
|
124 |
+
# Function to toggle visibility
|
125 |
+
def switch_output(selected_output):
|
126 |
+
if selected_output == "Chat History":
|
127 |
+
return gr.update(visible=True), gr.update(visible=False)
|
128 |
+
elif selected_output == "EHR Details":
|
129 |
+
return gr.update(visible=False), gr.update(visible=True)
|
130 |
+
|
131 |
+
# Set up the change event
|
132 |
+
output_selector.change(
|
133 |
+
fn=switch_output,
|
134 |
+
inputs=[output_selector],
|
135 |
+
outputs=[chat_history_output, ehr_details_output], # Update visibility of both components
|
136 |
+
)
|
137 |
+
|
138 |
+
notification_box = gr.Textbox(label="Error", interactive=False, lines=2)
|
139 |
|
140 |
# Bind Get Nurse Response button
|
141 |
send_button.click(
|
|
|
151 |
chat_history_button = gr.Button("View Chat History")
|
152 |
ehr_details_button = gr.Button("View EHR Details")
|
153 |
with gr.Column():
|
154 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
155 |
ehr_prompt_output = gr.Textbox(
|
156 |
label="Outputs",
|
157 |
interactive=False,
|
|
|
176 |
)
|
177 |
|
178 |
send_button.click(
|
|
|
179 |
fn=view_ehr_details,
|
180 |
inputs=[gr.Textbox(value="details", visible=False)],
|
181 |
outputs=ehr_details_output
|
182 |
)
|
183 |
|
184 |
send_button.click(
|
|
|
185 |
fn=view_ehr_details,
|
186 |
inputs=[gr.Textbox(value="prompt", visible=False)],
|
187 |
outputs=ehr_prompt_output
|