Chat Interfaces

app.py

@@ -0,0 +1,60 @@
+import random
+import time
+import gradio as gr
+
+def placeholder(input, history):
+    return "You typed: " + input
+
+class ChatbotInterface():
+    def __init__(self, name):
+        self.name = name
+        self.chatbot = gr.Chatbot()
+        self.chat_history = []
+        
+        with gr.Row() as row:
+            row.justify = "end"
+            self.msg = gr.Textbox(scale=7)
+            self.submit = gr.Button("Submit", scale=1)
+
+        clear = gr.ClearButton([self.msg, self.chatbot])
+        chat_history = []
+        
+        self.submit.click(self.respond, [self.msg, self.chatbot], [self.msg, self.chatbot])
+    
+    def respond(self, msg, history):
+            bot_message = random.choice(["Hello, I'm MedChat! How can I help you?", "Hello there! I'm Medchat, a medical assistant! How can I help you?"])
+            self.chat_history.append([msg, bot_message])
+            time.sleep(1)
+            return "", self.chat_history
+
+if __name__ == "__main__":
+    with gr.Blocks() as demo:
+        with gr.Row() as intro:
+            gr.Markdown(
+                """
+                ## MedChat
+                Welcome to MedChat, a medical assistant chatbot! You can currently chat with three chatbots that are trained on the same medical dataset.
+                
+                If you want to compare the output of each model, click the submit to all button and see the magic happen!
+                """
+            )
+        with gr.Row() as row:
+            with gr.Column() as col1:
+                with gr.Tab("GaiaMinimed") as gaia:
+                    gaia_bot = ChatbotInterface("GaiaMinimed")
+            with gr.Column() as col2:
+                with gr.Tab("MistralMed") as mistral:
+                    mistral_bot = ChatbotInterface("MistralMed") 
+                with gr.Tab("Falcon-7B") as falcon7b:
+                    falcon_bot = ChatbotInterface("Falcon-7B")
+
+
+        def submit_to_all(value):
+            for element in [gaia_bot, mistral_bot, falcon_bot]:
+                element.msg.value = value
+                element.submit.click()
+                
+        submit_all = gr.Button("Submit to All", scale=1)
+        submit_all.click(submit_to_all, [gaia_bot.msg])
+                
+    demo.launch()