Update app.py

app.py

@@ -15,7 +15,6 @@ def csv_prompter(question):
     json_data = json.load(json_file)
     json_data = json_data['records']
     
-    print(json_data)
     fulltext = []
     #print all nm in json file
     for i in json_data:
@@ -40,9 +39,8 @@ def csv_prompter(question):
     #fulltext = fulltext + "\n 00:00 - Customer: Hey, dear, can I please take out? 00:03 - Waiter: I'm up here. Okay go ahead. 00:06 - Customer: Can I get two sesame chicken dinners? 00:10 - Customer: One with lo mein and one with fried rice? 00:13 - Customer: Can I get, I know spring rolls come with each of those, but can I get just as long 00:17 - Customer: as I've got a total of two spring rolls and two egg rolls? 00:21 - Waiter: Okay. Anything? 00:22 - Customer: And a small bowl of hot and sour soup. 00:25 - Customer: And that's it. 00:26 - Waiter: Alright. Two sesame chickens, small one with fried rice, small one with lo mein, and both of those with spring rolls, two egg rolls and a small hot and sour soup. 00:30 - Customer: It's 205-473-1750. 00:36 - Waiter: What's the fun number please? 00:43 - Waiter: Just got around 20 minutes. Thank you. 00:43 - Customer: Okay, dear, thank you so much."
     fulltext = fulltext+'\n'
     
-    fulltext = fulltext+'\n'+question+'\n'
+    fulltext = fulltext+'\n\n'+question
     
-    fulltext = fulltext+'Based on the above dialogue and menu, If you were a waiter, what would you say next?\n Waiter:'
     print(fulltext)
     messages = [
             {"role": "system", "content": ""},
@@ -73,7 +71,7 @@ with gr.Blocks() as demo:
     def restart(history):
         invitation = "Waiter: "
         human_invitation = "Customer: "
-        return [[" \n",invitation+"Taste of China, may I help you?"]]
+        return [["\n",invitation+"Taste of China, may I help you?"]]
             
     def user(user_message, history):
         invitation = "Waiter: "
@@ -84,13 +82,12 @@ with gr.Blocks() as demo:
         invitation = "Waiter: "
         human_invitation = "Customer: "
         print(history)
-
         
-        question = ""
-        for each_ques in history:
-            question = question+ each_ques[0].replace("Customer: ","")+" \n"
+        alist  = list(itertools.chain.from_iterable(history))
+        alist=filter(None, alist)
+        fulltext = "Based on the above menu and following dialogue, If you were a waiter, what would you say next?\n"+ "\n".join(alist) + "\n" + invitation
 
-        response = csv_prompter(question)
+        response = csv_prompter(fulltext)
         
         response = invitation+ response
         history[-1][1] = response