Update app.py

app.py

@@ -21,8 +21,8 @@ def speak(text):
     # Do not try to play the audio, just save it
     # os.system("mpg321 output.mp3")  # Removed, as it is not supported in Hugging Face
 
-# Function to process the order
-def process_order(order):
+# Function to process the order and handle confirmation
+def process_order(order, confirmation):
     response = "You have ordered the following: "
     order = order.lower()
 
@@ -35,18 +35,25 @@ def process_order(order):
 
     if ordered_items:
         response += ', '.join(ordered_items) + ". Is that correct?"
-        speak(response)
-        return response
+        if confirmation.lower() == "yes":
+            speak("Thank you for your order. It will be ready shortly!")
+            return "Order confirmed. " + response
+        else:
+            speak("Please tell me again what you'd like to order.")
+            return "Order not confirmed. Please try again."
     else:
         speak("Sorry, I couldn't find any items matching your order. Can you try again?")
         return "Sorry, I couldn't find any items matching your order. Can you try again?"
 
 # Create Gradio interface
-def start_assistant(order):
-    return process_order(order)
+def start_assistant(order, confirmation):
+    return process_order(order, confirmation)
 
 # Gradio interface setup
-iface = gr.Interface(fn=start_assistant, inputs=gr.Textbox(label="Type your order"), outputs="text")
+iface = gr.Interface(fn=start_assistant, inputs=["text", "text"], outputs="text", 
+                     live=True, 
+                     title="Voice Food Ordering Assistant", 
+                     description="Type your food order and confirm your order.")
 
 # Launch the interface
 iface.launch()