AIVoice

Sleeping

App Files Files Community

geethareddy commited on Jan 3, 2025

Commit

2208db2

verified ·

1 Parent(s): 6dcad94

Update app.py

Browse files

Files changed (1) hide show

app.py +55 -104

app.py CHANGED Viewed

@@ -6,62 +6,20 @@ import ffmpeg
 import logging
 from werkzeug.exceptions import BadRequest
-# Initialize Flask App
 app = Flask(__name__)
-# Set up logging
 logging.basicConfig(level=logging.INFO)
-# Menu details
 MENU = {
-    "Appetizer": {
-        "Veg Samosas": 3,
-        "Pakodas Onion": 9,
-        "Pakodas Spinach": 9,
-        "Pakodas Mixed Vegetable": 9,
-    },
-    "Manchurian": {
-        "Vegetable": 12,
-        "Gobi (Cauliflower florets)": 12,
-        "Paneer (Cottage Cheese)": 13,
-        "Chicken": 14,
-        "Fish": 15,
-        "Jhinga (Shrimp)": 17,
-    },
-    "Main Course Veg Section": {
-        "Dal Fry": 11,
-        "Dal Makhani": 12,
-        "Channa Masala": 12,
-        "Aloo Gobi Masala": 12,
-        "Saag Paneer": 12,
-        "Kadai Paneer": 12,
-        "Paneer Butter Masala": 12,
-        "Paneer Tikka Masala": 12,
-        "Dum Ka Paneer": 12,
-    },
-    "Main Course Non-Veg Section": {
-        "Chicken Saag": 15,
-        "Gosht (Goat) Saag": 16,
-        "Chicken Curry": 15,
-        "Gosht (Goat) Curry": 16,
-        "Butter Chicken": 15,
-        "Tikka Masala Chicken": 15,
-    },
-    "Biryani": {
-        "Vegetable": 12,
-        "Egg": 12,
-        "Paneer": 14,
-        "Chicken Dum (Bone In)": 14,
-        "Special Chicken (Boneless)": 15,
-        "Shrimp": 17,
-        "Kheema (Goat)": 17,
-    },
 }
-# Cart for storing orders
-cart = []
-# HTML Template for Frontend
 html_code = """
 <!DOCTYPE html>
 <html lang="en">
@@ -72,13 +30,8 @@ html_code = """
     <style>
         body {
             font-family: Arial, sans-serif;
             background-color: #f4f4f9;
-            display: flex;
-            flex-direction: column;
-            align-items: center;
-            justify-content: center;
-            height: 100vh;
-            margin: 0;
         }
         h1 {
             color: #333;
@@ -91,37 +44,17 @@ html_code = """
             color: white;
             font-size: 24px;
             border: none;
-            display: flex;
-            align-items: center;
-            justify-content: center;
             cursor: pointer;
-            box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1);
-            transition: background-color 0.3s;
-        }
-        .mic-button:hover {
-            background-color: #0056b3;
         }
-        .status {
             margin-top: 20px;
-            font-size: 18px;
-            color: #666;
-        }
-        .response {
-            margin-top: 20px;
-            padding: 10px;
-            background-color: #fff;
-            border: 1px solid #ddd;
-            border-radius: 5px;
-            box-shadow: 0px 4px 6px rgba(0, 0, 0, 0.1);
-            width: 300px;
-            text-align: center;
         }
     </style>
 </head>
 <body>
-    <h1>BIRYANI HUB</h1>
     <button class="mic-button" id="mic-button">🎤</button>
-    <div class="status" id="status">Press the mic button to start the conversation...</div>
     <div class="response" id="response" style="display: none;">Response will appear here...</div>
     <script>
         const micButton = document.getElementById('mic-button');
@@ -137,8 +70,12 @@ html_code = """
         });
         function startConversation() {
-            status.textContent = 'Listening...';
-            startListening();
         }
         function startListening() {
@@ -161,8 +98,8 @@ html_code = """
                         const utterance = new SpeechSynthesisUtterance(data.response);
                         speechSynthesis.speak(utterance);
                     } catch (error) {
-                        response.textContent = 'Error occurred. Please try again.';
-                        status.textContent = 'Press the mic button to restart.';
                         isConversationActive = false;
                     }
                 };
@@ -192,6 +129,9 @@ def process_audio():
         temp_file = NamedTemporaryFile(delete=False, suffix=".webm")
         audio_file.save(temp_file.name)
         converted_file = NamedTemporaryFile(delete=False, suffix=".wav")
         ffmpeg.input(temp_file.name).output(
             converted_file.name, acodec='pcm_s16le', ac=1, ar='16000'
@@ -200,37 +140,48 @@ def process_audio():
         recognizer = sr.Recognizer()
         with sr.AudioFile(converted_file.name) as source:
             audio_data = recognizer.record(source)
-            command = recognizer.recognize_google(audio_data).lower()
-            response = process_command(command)
         return jsonify({"response": response})
     except Exception as e:
-        return jsonify({"response": f"Error: {str(e)}"})
     finally:
-        if os.path.exists(temp_file.name):
-            os.unlink(temp_file.name)
-        if os.path.exists(converted_file.name):
-            os.unlink(converted_file.name)
 def process_command(command):
     global cart
     if "menu" in command:
-        menu_items = ". ".join(f"{category}: {', '.join(items.keys())}" for category, items in MENU.items())
-        return f"Our menu includes: {menu_items}. What would you like to order?"
-    elif "order" in command:
-        for category, items in MENU.items():
-            for item, price in items.items():
-                if item.lower() in command:
-                    cart.append((item, price))
-                    return f"{item} has been added to your cart. Anything else?"
-    elif "that's it" in command:
         if cart:
-            total_price = sum(item[1] for item in cart)
-            items = ", ".join(f"{item[0]} (${item[1]})" for item in cart)
-            return f"Your cart includes: {items}. The total price is ${total_price}. Thank you for your order!"
         else:
-            return "Your cart is empty. Thank you!"
-    return "Sorry, I didn't understand. Please ask for the menu or place an order."
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)

 import logging
 from werkzeug.exceptions import BadRequest
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
+# Global cart to store items and their prices
+cart = []
+# Menu with items and their prices
 MENU = {
+    "biryani": {"chicken biryani": 150, "veg biryani": 120},
+    "starters": {"chicken wings": 180, "paneer tikka": 160},
+    "breads": {"butter naan": 40, "roti": 30},
+    "curries": {"butter chicken": 200, "dal fry": 150}
 }
 html_code = """
 <!DOCTYPE html>
 <html lang="en">
     <style>
         body {
             font-family: Arial, sans-serif;
+            text-align: center;
             background-color: #f4f4f9;
         }
         h1 {
             color: #333;
             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');
         });
         function startConversation() {
+            const utterance = new SpeechSynthesisUtterance('Please give your order.');
+            speechSynthesis.speak(utterance);
+            utterance.onend = () => {
+                status.textContent = 'Listening...';
+                startListening();
+            };
         }
         function startListening() {
                         const utterance = new SpeechSynthesisUtterance(data.response);
                         speechSynthesis.speak(utterance);
                     } catch (error) {
+                        response.textContent = 'Error processing your request. Please try again.';
+                        status.textContent = 'Error occurred. Please press the mic button to restart.';
                         isConversationActive = false;
                     }
                 };
         temp_file = NamedTemporaryFile(delete=False, suffix=".webm")
         audio_file.save(temp_file.name)
+        if os.path.getsize(temp_file.name) == 0:
+            raise BadRequest("Uploaded audio file is empty.")
         converted_file = NamedTemporaryFile(delete=False, suffix=".wav")
         ffmpeg.input(temp_file.name).output(
             converted_file.name, acodec='pcm_s16le', ac=1, ar='16000'
         recognizer = sr.Recognizer()
         with sr.AudioFile(converted_file.name) as source:
             audio_data = recognizer.record(source)
+            try:
+                command = recognizer.recognize_google(audio_data)
+                response = process_command(command)
+            except sr.UnknownValueError:
+                response = "Sorry, I could not understand. Please try again."
         return jsonify({"response": response})
+    except BadRequest as br:
+        return jsonify({"response": f"Bad Request: {str(br)}"}), 400
     except Exception as e:
+        return jsonify({"response": f"An error occurred: {str(e)}"}), 500
     finally:
+        os.unlink(temp_file.name)
+        os.unlink(converted_file.name)
 def process_command(command):
     global cart
+    command = command.lower()
+    all_items = {item: price for category in MENU.values() for item, price in category.items()}
     if "menu" in command:
+        menu = ", ".join([f"{category}: {', '.join(items.keys())}" for category, items in MENU.items()])
+        return f"Here is our menu: {menu}."
+    elif any(item in command for item in all_items.keys()):
+        item = next((item for item in all_items.keys() if item in command), None)
+        if item:
+            cart.append((item, all_items[item]))
+            total = sum(item[1] for item in cart)
+            return f"{item.capitalize()} added to your cart. Your current total is ₹{total}."
+    elif "final order" in command or "submit" in command:
         if cart:
+            items = ", ".join([f"{item[0]} (₹{item[1]})" for item in cart])
+            total = sum(item[1] for item in cart)
+            cart.clear()  # Reset cart after final order
+            return f"Your final order is: {items}. The total price is ₹{total}. Thank you for ordering!"
         else:
+            return "Your cart is empty. Please add items first."
+    elif "no" in command or "nothing" in command or "goodbye" in command:
+        cart.clear()
+        return "Goodbye! Thank you for using AI Dining Assistant."
+    return "Sorry, I didn't understand that. Please try again."
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)