clear btn and dropdown

app.py

@@ -1,16 +1,19 @@
 from pathlib import Path
 
 from dotenv import load_dotenv
+
 load_dotenv()
 
-from langchain.chains import LLMChain
-from langchain.prompts import PromptTemplate
-from langchain.chat_models import ChatOpenAI
+import json
+import os
+from functools import partial
+from pathlib import Path
+
 import gradio as gr
 from elevenlabs import generate, play
-from pathlib import Path
-import json
-import os 
+from langchain.chains import LLMChain
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts import PromptTemplate
 
 # import whisper
 
@@ -29,39 +32,49 @@ chain = LLMChain(llm=llm, prompt=prompt)
 with open("data/patients.json") as f:
     patiens = json.load(f)
 
-patient = patiens[0]
+patients_names = [el["name"] for el in patiens]
+patient = [patiens[0]]
 
-print(patient)
 
-def run_text_prompt(message, chat_history):
-    bot_message = chain.run(patient=patient, user_input=message)
-
-    # audio = generate(text=bot_message, voice="Bella")
-
-    # play(audio, notebook=True)
+def run_text_prompt(message, chat_history, patient):
+    print(message, chat_history, patient[0]["name"])
+    bot_message = chain.run(patient=patient[0], user_input=message)
 
     chat_history.append((message, bot_message))
     return "", chat_history
 
 
-# def run_audio_prompt(audio, chat_history):
-#     if audio is None:
-#         return None, chat_history
-
-#     message_transcription = model.transcribe(audio)["text"]
-#     _, chat_history = run_text_prompt(message_transcription, chat_history)
-#     return None, chat_history
+def on_drop_down_change(selected_item):
+    index = patients_names.index(selected_item)
+    patient[0] = patiens[index]
+    print(f"You selected: {selected_item}", index)
+    return f"```json\n{json.dumps(patient[0], indent=2)}\n```"
 
 
 with gr.Blocks() as demo:
-    gr.Markdown(f"```json\n{json.dumps(patient)}```")
     chatbot = gr.Chatbot()
 
-    msg = gr.Textbox()
-    msg.submit(run_text_prompt, [msg, chatbot], [msg, chatbot])
-    
-    # with gr.Row():
-        # audio = gr.Audio(source="microphone", type="filepath")
-        
-        # send_audio_button = gr.Button("Send Audio", interactive=True)
-        # send_audio_button.click(run_audio_prompt, [audio, chatbot], [audio, chatbot])
+    with gr.Row():
+        with gr.Column():
+            msg = gr.Textbox()
+            msg.submit(
+                partial(run_text_prompt, patient=patient),
+                [msg, chatbot],
+                [msg, chatbot],
+            )
+            clear = gr.Button("Clear")
+            clear.click(lambda: None, None, chatbot, queue=False)
+
+        with gr.Column():
+            patients_names = [el["name"] for el in patiens]
+            dropdown = gr.Dropdown(
+                choices=patients_names,
+                value=patients_names[0],
+                interactive=True,
+                label="Patient",
+            )
+            markdown = gr.Markdown(f"```json\n{json.dumps(patient[0], indent=2)}\n```")
+            dropdown.change(fn=on_drop_down_change, inputs=dropdown, outputs=markdown)
+            print(patient)
+            # dropdown.value(lambda x: print(x))
+            # patient = dropdown.select(on_dropdown_select)