Update app.py

app.py

@@ -48,27 +48,48 @@ def reset_state():
     return [], []
 
 
-with gr.Blocks() as demo:
-    gr.HTML("""<h1 align="center">So Rude</h1>""")
+def AIPatient(message):
+
+    global isFirstRun, history,context
+
+    if isFirstRun:
+      context = initContext
+      isFirstRun = False
+    #else:
+      #for turn in history:
+      #  context += f"\n<|im_start|> Nurse: {turn[0]}\n<|im_start|> Barry: {turn[1]}"
+    context += """
+                  <|im_start|>nurse
+                  Nurse: """+message+"""
+                  <|im_start|>barry
+                  Barry:
+                  """
 
-    chatbot = gr.Chatbot()
-    with gr.Row():
-        with gr.Column(scale=4):
-            user_input = gr.Textbox(show_label=False, placeholder="Input...", lines=8, elem_id="user_input")
-            submitBtn = gr.Button("Submit", variant="primary", elem_id="submit_btn")
-        with gr.Column(scale=1):
-            max_length = gr.Slider(0, 256, value=64, step=1.0, label="Maximum Length", interactive=True)
-            top_p = gr.Slider(0, 1, value=0.7, step=0.01, label="Top P", interactive=True)
-            temperature = gr.Slider(0, 2.0, value=0.95, step=0.01, label="Temperature", interactive=True)
-            emptyBtn = gr.Button("Clear History")
+    response = ""
+    # Here, you should add the code to generate the response using your model
+    # For example:
+    while(len(response) < 1):
+      output = llm(context, max_tokens=400, stop=["Nurse:"], echo=False)
+      response = output["choices"][0]["text"]
+      response = response.strip()
 
-    history = gr.State([])
+    with feedback_file.open("a") as f:
+            f.write(json.dumps({"Nurse": message, "Barry": response},indent=4))
+            f.write("\n")
 
-    submitBtn.click(
-        predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history], show_progress=True
-    )
-    submitBtn.click(reset_user_input, [], [user_input])
+    context += response
+    print (context)
 
-    emptyBtn.click(reset_state, outputs=[chatbot, history], show_progress=True)
+    history.append((message,response))
+    return history
 
-demo.queue().launch(share=False, inbrowser=True)
+with gr.Blocks() as demo:
+    gr.Markdown("# AI Patient Chatbot")
+    with gr.Group():
+        with gr.Tab("Patient Chatbot"):
+            chatbot = gr.Chatbot()
+            message = gr.Textbox(label="Enter your message to Barry", placeholder="Type here...", lines=2)
+            send_message = gr.Button("Submit")
+            send_message.click(AIPatient, inputs=[message], outputs=[chatbot])
+            save_chatlog = gr.Button("Save Chatlog")
+            #send_message.click(SaveChatlog, inputs=[message], outputs=[chatbot])