Update app.py

app.py

@@ -8,11 +8,9 @@ openai_api_key_textbox = ""
 model = None
 tokenizer = None
 generator = None
-csv_name = "disease_database_mini.csv"
-df = pd.read_csv(csv_name)
 openai.api_key = "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"
 
-def csv_prompter(question,csv_name):    
+def csv_prompter(question):    
     json_file = open('order.json')
     json_data = json.load(json_file)
     json_data = json_data['records']
@@ -48,7 +46,7 @@ def csv_prompter(question,csv_name):
     
     
     
-    fulltext = fulltext+'Based on the above dialogue and menu, output the dishes ordered by the customer. Note that some dishes have sub-options (e.g. some are set meals, some are a la carte)'
+    fulltext = fulltext+'Based on the above dialogue and menu, If you were a waiter, what would you answer next?'
     print(fulltext)
     messages = [
             {"role": "system", "content": ""},
@@ -65,14 +63,11 @@ def csv_prompter(question,csv_name):
     
     print(response)
 
-    return answer_llm
+    return response
 
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Autonomous ChatDoctor (openai version), based on disease database knowledge")
-    gr.Markdown("## Example: If I have frontal headache, fever, and painful sinuses, what disease should I have, and what medical test should I take?")
-    gr.Markdown("Our model will answer based on the content of the excel below, so please try to ask questions based on the table content.")
     
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
@@ -80,26 +75,26 @@ with gr.Blocks() as demo:
     Initialization = gr.Button("Initialization")
 
     def restart(history):
-        invitation = "ChatDoctor: "
-        human_invitation = "Patient: "
-        return [[" \n",invitation+" I am ChatDoctor, what medical questions do you have?"]]
+        invitation = "Waiter: "
+        human_invitation = "Customer: "
+        return [[" \n",invitation+"Taste of China, may I help you?"]]
             
     def user(user_message, history):
-        invitation = "ChatDoctor: "
-        human_invitation = "Patient: "
+        invitation = "Waiter: "
+        human_invitation = "Customer: "
         return "", history +[[human_invitation+user_message, None]]
 
     def bot(history):
-        invitation = "ChatDoctor: "
-        human_invitation = "Patient: "
+        invitation = "Waiter: "
+        human_invitation = "Customer: "
         print(history)
 
         
         question = ""
         for each_ques in history:
-            question = question+ each_ques[0].replace("Patient: ","")+" \n"
+            question = question+ each_ques[0].replace("Customer: ","")+" \n"
 
-        response = csv_prompter(question,csv_name)
+        response = csv_prompter(question)
         
         response = invitation+ response
         history[-1][1] = response