Spaces:
Sleeping
Sleeping
File size: 7,530 Bytes
7677cff 36f07df 7677cff 8a4e542 7677cff adbdaee 36f07df adbdaee 7677cff 36f07df 7677cff 70e2745 7677cff 70e2745 7677cff 70e2745 7677cff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 |
import gradio as gr
import requests
API_BASE_URL = "http://localhost:8000" # Update this with your actual API base URL if deployed elsewhere
# Function: Get nurse response
def get_nurse_response(user_input, model_name, chat_history):
try:
start_time = time.time()
# Send user input to the API
response = requests.post(
f"{API_BASE_URL}/nurse_response",
json={"user_input": user_input, "model_name": model_name},
timeout=15
)
response.raise_for_status()
end_time = time.time()
elapsed_time = end_time - start_time
nurse_response = response.json().get("nurse_response", "No response received.")
# Append user and nurse messages to the chat history
chat_history.append((f"👤 {user_input}", f"🤖 {nurse_response} ({elapsed_time:.2f} s.)"))
return chat_history, "" # Clear the user input after sending
except requests.exceptions.RequestException as e:
chat_history.append(("⚠️ Error", str(e)))
return chat_history, ""
# Function: Reset chat history
def reset_history():
response = requests.post(f"{API_BASE_URL}/reset")
return [], "", response.text
# Function: View chat history
def view_chat_history():
try:
response = requests.get(f"{API_BASE_URL}/history")
response.raise_for_status()
chat_history = response.json().get("chat_history", [])
if not chat_history:
return "No chat history available."
# Properly format chat history for display
formatted_history = []
for message in chat_history:
if message.get("role") == "user":
formatted_history.append(f"👤 User: {message.get('content', '')}")
elif message.get("role") == "assistant":
formatted_history.append(f"🤖 Nurse: {message.get('content', '')}")
return "\n".join(formatted_history)
except requests.exceptions.RequestException as e:
return f"Error: {str(e)}"
import json
import time
# Function: View EHR details
def view_ehr_details(view):
try:
time.sleep(2.5)
response = requests.get(f"{API_BASE_URL}/details")
response.raise_for_status()
ehr_data = response.json()
if view == "details":
ehr_data = ehr_data["ehr_data"]
elif view == "prompt":
ehr_data.pop("ehr_data", None)
ehr_data = json.dumps(ehr_data, indent=4, ensure_ascii=False)
return ehr_data
except requests.exceptions.RequestException as e:
return f"Error: {str(e)}"
# Gradio Interface
def create_gradio_interface():
with gr.Blocks() as interface:
# Title and description
gr.Markdown(
"""
# MALI_NURSE Gradio Interface
### A User-Friendly Interface to Interact with the MALI_NURSE API
Select a model, input your question, and view nurse responses or manage chat history and EHR details.
"""
)
# Main Input Section
with gr.Row():
with gr.Column(scale=2):
chat_box = gr.Chatbot(label="Chat with MALI Nurse", scale=1)
send_button = gr.Button("Send", variant="primary", size="lg", scale=1)
with gr.Row():
user_input = gr.Textbox(
label="Your Message",
placeholder="Type your question or message here...",
lines=2,
)
model_name = gr.Radio(
choices=["typhoon-v1.5x-70b-instruct", "openthaigpt", "llama-3.3-70b-versatile"],
value="typhoon-v1.5x-70b-instruct",
label="Model Selection",
)
with gr.Column(scale=1):
output_selector = gr.Dropdown(
choices=["Chat History", "EHR Details"],
value="Chat History",
label="Select Output to Display",
)
chat_history_output = gr.Textbox(
label="Chat History Output",
interactive=False,
lines=6,
scale=1,
visible=True, # Initially visible
)
ehr_details_output = gr.Textbox(
label="EHR Details Output",
interactive=False,
lines=6,
scale=1,
visible=False, # Initially hidden
)
# Function to toggle visibility
def switch_output(selected_output):
if selected_output == "Chat History":
return gr.update(visible=True), gr.update(visible=False)
elif selected_output == "EHR Details":
return gr.update(visible=False), gr.update(visible=True)
# Set up the change event
output_selector.change(
fn=switch_output,
inputs=[output_selector],
outputs=[chat_history_output, ehr_details_output], # Update visibility of both components
)
notification_box = gr.Textbox(label="Error", interactive=False, lines=2)
# Bind Get Nurse Response button
send_button.click(
fn=get_nurse_response,
inputs=[user_input, model_name, chat_box],
outputs=[chat_box, user_input], # Update chat box and clear input
)
# Advanced Options
with gr.Accordion("Advanced Options", open=False):
with gr.Row():
reset_button = gr.Button("Reset Data", variant="primary")
chat_history_button = gr.Button("View Chat History")
ehr_details_button = gr.Button("View EHR Details")
with gr.Column():
ehr_prompt_output = gr.Textbox(
label="Outputs",
interactive=False,
lines=6,
)
# Bind buttons to respective functions
reset_button.click(
fn=reset_history,
inputs=[],
outputs=[chat_box, user_input, notification_box], # Clear chat box and input
)
chat_history_button.click(
fn=view_chat_history,
inputs=[],
outputs=chat_history_output,
)
send_button.click(
fn=view_chat_history,
inputs=[],
outputs=chat_history_output,
)
send_button.click(
fn=view_ehr_details,
inputs=[gr.Textbox(value="details", visible=False)],
outputs=ehr_details_output
)
send_button.click(
fn=view_ehr_details,
inputs=[gr.Textbox(value="prompt", visible=False)],
outputs=ehr_prompt_output
)
# Footer
gr.Markdown(
"""
---
Developed by **Piang** 🚀
Powered by Typhoon v1.5x and OpenThaiGPT Models.
"""
)
return interface
# Run the Gradio Interface
if __name__ == "__main__":
gr_interface = create_gradio_interface()
gr_interface.launch(server_name="0.0.0.0", server_port=7860)
|