Update app.py

app.py

@@ -6,63 +6,62 @@ import google.generativeai as palm
 palm.configure(api_key=os.environ.get("palm_key"))
 
 defaults = {
-  'model': 'models/chat-bison-001',
-  'temperature': 0.3,
-  'candidate_count': 1,
-  'top_k': 40,
-  'top_p': 0.95,
+    'model': 'models/chat-bison-001',
+    'temperature': 0.3,
+    'candidate_count': 1,
+    'top_k': 40,
+    'top_p': 0.95,
 }
 context = "You are an order bot with only 3 types of pizzas: Margherita, Cheesy-Bacon and Vegetarian. Prices are small (9$), medium (11$) and large (13$). You can offer 2 types of drinks coke and mountain dew."
 examples = [
-  [
-    "Hi, I want to order pizzas.",
-    "Hello, we have 3 types of pizzas: Margherita, Cheesy-Bacon and Vegetarian. Prices are small (9$), medium (11$) and large (13$)."
-  ],
-  [
-    "I want a large margherita and a medium cheesy-bacon.",
-    "Do you want any drinks? We have coke (3$) or Mountain dew (5$)."
-  ],
-  [
-    "Yes 2 cokes.",
-    "Total will be 19$. Please enter your address and phone number"
-  ],
-  [
-    "Paris eiffel tower, 0653145863",
-    "Perfect ! Do you need anything else?"
-  ],
-  [
-    "No thanks.",
-    "Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"L\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
-  ],
-  [
-    "Yes, i'd like to change margherita size to M.",
-    "Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"M\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
-  ],
-  [
-    "No thanks",
-    "Have a nice day!"
-  ]
+    [
+        "Hi, I want to order pizzas.",
+        "Hello, we have 3 types of pizzas: Margherita, Cheesy-Bacon and Vegetarian. Prices are small (9$), medium (11$) and large (13$)."
+    ],
+    [
+        "I want a large margherita and a medium cheesy-bacon.",
+        "Do you want any drinks? We have coke (3$) or Mountain dew (5$)."
+    ],
+    [
+        "Yes 2 cokes.",
+        "Total will be 19$. Please enter your address and phone number"
+    ],
+    [
+        "Paris eiffel tower, 0653145863",
+        "Perfect ! Do you need anything else?"
+    ],
+    [
+        "No thanks.",
+        "Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"L\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
+    ],
+    [
+        "Yes, i'd like to change margherita size to M.",
+        "Have a nice day!\nHere is the sum up: \n{\npizza_type_1: \"margherita\",\npizza_type_2: \"cheesy-bacon\",\npizza_size_1: \"M\",\npizza_size_2: \"M\",\ndrink_1: \"coke\",\ndrink_2: \"coke\",\ntotal_price: \"19\",\naddress: \"Paris, Eiffel Tower\",\nphone_number: \"0653145863\"\n}\nDo you want to change anything?"
+    ],
+    [
+        "No thanks",
+        "Have a nice day!"
+    ]
 ]
 
-history = ['']
+history = [['']]
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
     btn = gr.Button("Submit", variant="primary")
-    # gr.ClearButton([msg, chatbot])
     clear = gr.Button("Clear")
-   
+    
     def user(user_message, history):
-        return gr.update(value=""), history + [[user_message, None]]
+        history.append([user_message, None])
+        return gr.update(value=""), history
     
     def bot(history):
-        try:        
+        try:
             bot_message = palm.chat(
                 context=context,
                 examples=examples,
-                messages=history[-1][0]
-                # messages=history
+                messages=[h[0] for h in history]
             )
             
             history[-1][1] = ""
@@ -89,4 +88,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     clear.click(lambda: None, None, chatbot, queue=False)
 
 demo.queue()
-demo.launch(debug=True)
+demo.launch()