Update app.py

app.py

@@ -25,7 +25,7 @@ def generate_voice_response(text):
     return temp_file.name
 
 def calculate_total(cart):
-    return len(cart)  # Only count items, not prices
+    return sum(menu_items[item] for item in cart)
 
 def restaurant_voice_assistant(audio, state_json):
     global cart
@@ -83,10 +83,11 @@ def restaurant_voice_assistant(audio, state_json):
         response += "\nWhat would you like to add to your cart or ask about?"
     elif "final order" in input_text or "submit order" in input_text:
         if cart:
+            total = calculate_total(cart)
             response = "Your final order includes:\n"
             for item in cart:
                 response += f"- {item}\n"
-            response += f"\nThank you for ordering!"
+            response += f"\nTotal amount: ${total:.2f}.\nThank you for ordering!"
             cart = []  # Clear cart after finalizing order
             state["current_items"] = []  # Clear current cycle tracking
         else:
@@ -101,13 +102,13 @@ with gr.Blocks() as demo:
     state = gr.State(value=json.dumps({}))
 
     with gr.Row():
-        user_audio = gr.Audio(type="filepath", label="Your Voice Input")
-        output_text = gr.Textbox(label="Response Text")
+        record_button = gr.Audio(source="microphone", type="filepath", label="Speak Now")
+        output_text = gr.Textbox(label="Response Text", interactive=False)
 
     with gr.Row():
         voice_output = gr.Audio(label="Response Audio", autoplay=True)
 
     # Automatically process audio when recording stops
-    user_audio.change(restaurant_voice_assistant, inputs=[user_audio, state], outputs=[output_text, voice_output, state])
+    record_button.change(restaurant_voice_assistant, inputs=[record_button, state], outputs=[output_text, voice_output, state])
 
 demo.launch()