Spaces:

geethareddy
/

voice1

Sleeping

App Files Files Community

geethareddy commited on Jan 16, 2025

Commit

1194ded

verified ·

1 Parent(s): 10671d0

Update app.py

Browse files

Files changed (1) hide show

app.py +18 -106

app.py CHANGED Viewed

@@ -23,107 +23,19 @@ current_category = None
 current_item = None
 awaiting_quantity = False
-# HTML Template for Frontend
-html_code = """
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>AI Dining Assistant</title>
-    <style>
-        body {
-            font-family: Arial, sans-serif;
-            text-align: center;
-            background-color: #f4f4f9;
-        }
-        h1 {
-            color: #333;
-        }
-        .mic-button {
-            width: 80px;
-            height: 80px;
-            border-radius: 50%;
-            background-color: #007bff;
-            color: white;
-            font-size: 24px;
-            border: none;
-            cursor: pointer;
-        }
-        .status, .response {
-            margin-top: 20px;
-        }
-    </style>
-</head>
-<body>
-    <h1>AI Dining Assistant</h1>
-    <button class="mic-button" id="mic-button">🎤</button>
-    <div class="status" id="status">Press the mic button to start...</div>
-    <div class="response" id="response" style="display: none;">Response will appear here...</div>
-    <script>
-        const micButton = document.getElementById('mic-button');
-        const status = document.getElementById('status');
-        const response = document.getElementById('response');
-        let isListening = false;
-        micButton.addEventListener('click', () => {
-            if (!isListening) {
-                isListening = true;
-                greetUser();
-            }
-        });
-        function greetUser() {
-            const utterance = new SpeechSynthesisUtterance("Hi. Welcome to Biryani Hub. Can I show you the menu?");
-            speechSynthesis.speak(utterance);
-            utterance.onend = () => {
-                status.textContent = "Listening...";
-                startListening();
-            };
-        }
-        async function startListening() {
-            const stream = await navigator.mediaDevices.getUserMedia({ audio: true });
-            const mediaRecorder = new MediaRecorder(stream, { mimeType: "audio/webm;codecs=opus" });
-            const audioChunks = [];
-            mediaRecorder.ondataavailable = (event) => audioChunks.push(event.data);
-            mediaRecorder.onstop = async () => {
-                const audioBlob = new Blob(audioChunks, { type: "audio/webm" });
-                const formData = new FormData();
-                formData.append("audio", audioBlob);
-                status.textContent = "Processing...";
-                try {
-                    const result = await fetch("/process-audio", { method: "POST", body: formData });
-                    const data = await result.json();
-                    response.textContent = data.response;
-                    response.style.display = "block";
-                    const utterance = new SpeechSynthesisUtterance(data.response);
-                    speechSynthesis.speak(utterance);
-                    utterance.onend = () => {
-                        if (!data.response.includes("Goodbye") && !data.response.includes("final order")) {
-                            startListening(); // Continue listening
-                        } else {
-                            status.textContent = "Conversation ended.";
-                            isListening = false;
-                        }
-                    };
-                } catch (error) {
-                    response.textContent = "Error processing your request. Please try again.";
-                    status.textContent = "Press the mic button to restart.";
-                    isListening = false;
-                }
-            };
-            mediaRecorder.start();
-            setTimeout(() => mediaRecorder.stop(), 5000); // Stop recording after 5 seconds
-        }
-    </script>
-</body>
-</html>
-"""
 @app.route("/")
 def index():
@@ -175,18 +87,18 @@ def process_command(command):
     command = command.lower()
     # Handle quantity input
-    if awaiting_quantity and command.isdigit():
-        quantity = int(command)
-        if quantity > 0:
             cart.append((current_item, MENU[current_category][current_item], quantity))
             awaiting_quantity = False
             item = current_item
             current_item = None
             total = sum(i[1] * i[2] for i in cart)
             cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
-            return f"Added {quantity} of {item} to your cart. Cart: {cart_summary}. Total: ₹{total}. Would you like to see the menu again?"
         else:
-            return "Quantity must be at least 1. How many would you like to order?"
     # Handle category selection
     for category, items in MENU.items():

 current_item = None
 awaiting_quantity = False
+# Extract quantity from command (e.g., "two" -> 2, "three" -> 3, etc.)
+def extract_quantity(command):
+    number_words = {
+        "one": 1, "two": 2, "three": 3, "four": 4, "five": 5,
+        "six": 6, "seven": 7, "eight": 8, "nine": 9, "ten": 10,
+        "1": 1, "2": 2, "3": 3, "4": 4, "5": 5, "6": 6, "7": 7, "8": 8, "9": 9, "10": 10
+    }
+    command_words = command.split()
+    for word in command_words:
+        if word in number_words:
+            return number_words[word]
+    return None
 @app.route("/")
 def index():
     command = command.lower()
     # Handle quantity input
+    if awaiting_quantity:
+        quantity = extract_quantity(command)
+        if quantity:
             cart.append((current_item, MENU[current_category][current_item], quantity))
             awaiting_quantity = False
             item = current_item
             current_item = None
             total = sum(i[1] * i[2] for i in cart)
             cart_summary = ", ".join([f"{i[0]} x{i[2]} (₹{i[1] * i[2]})" for i in cart])
+            return f"Added {quantity} x {item} to your cart. Your current cart: {cart_summary}. Total: ₹{total}. Would you like to add more items?"
         else:
+            return "Sorry, I couldn't understand the quantity. Please provide a valid quantity."
     # Handle category selection
     for category, items in MENU.items():