Update app.py

app.py

@@ -67,10 +67,11 @@ def AIPatient(message):
     response = ""
     # Here, you should add the code to generate the response using your model
     # For example:
-    for output in llm(input, stream=True, temperature=temperature, top_p=top_p, max_tokens=max_length, ):
-        piece = output['choices'][0]['text']
-        response += piece
-        chatbot[-1] = (chatbot[-1][0], response)
+    while(len(response) < 1):
+        print("here")
+      output = llm(context, max_tokens=400, stop=["Nurse:"], echo=False)
+      response = output["choices"][0]["text"]
+      response = response.strip()
 
     context += response
     print (context)
@@ -78,27 +79,19 @@ def AIPatient(message):
     history.append((message,response))
     return history
 
-with gr.Blocks() as demo:
-    gr.HTML("""<h1 align="center">So Rude</h1>""")
-
-    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")
 
-    history = gr.State([])
+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])
 
-    submitBtn.click(
-        predict, [user_input, chatbot, max_length, top_p, temperature, history], [chatbot, history], show_progress=True
-    )
-    submitBtn.click(reset_user_input, [], [user_input])
 
-    emptyBtn.click(reset_state, outputs=[chatbot, history], show_progress=True)
+            #message.submit(AIPatient, inputs=[message], outputs=[chatbot])
 
-demo.queue().launch(share=False, inbrowser=True)
+demo.launch(debug=True,share=False,inbrowser=True)