Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,33 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
21 |
config = AutoConfig.from_pretrained("openai/whisper-small")
|
22 |
config.update({"timeout": 60}) # Set timeout to 60 seconds
|
23 |
|
24 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
def generate_audio_prompt(text, filename):
|
26 |
try:
|
27 |
tts = gTTS(text)
|
@@ -32,31 +58,7 @@ def generate_audio_prompt(text, filename):
|
|
32 |
time.sleep(5) # Wait for 5 seconds before retrying
|
33 |
generate_audio_prompt(text, filename)
|
34 |
|
35 |
-
#
|
36 |
-
prompts = {
|
37 |
-
"welcome": "Welcome to Biryani Hub.",
|
38 |
-
"ask_name": "Tell me your name.",
|
39 |
-
"ask_email": "Please provide your email address.",
|
40 |
-
"thank_you": "Thank you for registration."
|
41 |
-
}
|
42 |
-
|
43 |
-
for key, text in prompts.items():
|
44 |
-
generate_audio_prompt(text, f"{key}.mp3")
|
45 |
-
|
46 |
-
# Symbol mapping for proper recognition
|
47 |
-
SYMBOL_MAPPING = {
|
48 |
-
"at the rate": "@",
|
49 |
-
"at": "@",
|
50 |
-
"dot": ".",
|
51 |
-
"underscore": "_",
|
52 |
-
"hash": "#",
|
53 |
-
"plus": "+",
|
54 |
-
"dash": "-",
|
55 |
-
"comma": ",",
|
56 |
-
"space": " "
|
57 |
-
}
|
58 |
-
|
59 |
-
# Function to convert audio to WAV format
|
60 |
def convert_to_wav(input_path, output_path):
|
61 |
try:
|
62 |
audio = AudioSegment.from_file(input_path)
|
@@ -66,45 +68,25 @@ def convert_to_wav(input_path, output_path):
|
|
66 |
print(f"Error: {str(e)}")
|
67 |
raise Exception(f"Audio conversion failed: {str(e)}")
|
68 |
|
69 |
-
# Function to check if audio contains actual speech
|
70 |
def is_silent_audio(audio_path):
|
71 |
audio = AudioSegment.from_wav(audio_path)
|
72 |
nonsilent_parts = detect_nonsilent(audio, min_silence_len=500, silence_thresh=audio.dBFS-16) # Reduced silence duration
|
73 |
print(f"Detected nonsilent parts: {nonsilent_parts}")
|
74 |
return len(nonsilent_parts) == 0 # If no speech detected
|
75 |
|
76 |
-
#
|
77 |
-
try:
|
78 |
-
print("Attempting to connect to Salesforce...")
|
79 |
-
sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
|
80 |
-
print("Connected to Salesforce successfully!")
|
81 |
-
print("User Info:", sf.UserInfo) # Log the user info to verify the connection
|
82 |
-
except Exception as e:
|
83 |
-
print(f"Failed to connect to Salesforce: {str(e)}")
|
84 |
-
|
85 |
-
# Function to fetch menu items from Salesforce
|
86 |
-
def get_menu_items():
|
87 |
-
query = "SELECT Name, Price__c, Ingredients__c, Category__c FROM Menu_Item__c"
|
88 |
-
result = sf.query(query)
|
89 |
-
return result['records']
|
90 |
-
|
91 |
-
# Define the route for the index page
|
92 |
@app.route("/")
|
93 |
def index():
|
94 |
return render_template("index.html")
|
95 |
|
96 |
-
|
97 |
def dashboard():
|
98 |
-
# You can pass user-specific data to this page if needed
|
99 |
return render_template("dashboard.html") # Render the dashboard template
|
100 |
|
101 |
-
# Function to create Salesforce record
|
102 |
-
# API endpoint to receive data from voice bot
|
103 |
@app.route('/login', methods=['POST'])
|
104 |
def login():
|
105 |
# Get data from voice bot (name, email, phone number)
|
106 |
data = request.json # Assuming voice bot sends JSON data
|
107 |
-
|
108 |
name = data.get('name')
|
109 |
email = data.get('email')
|
110 |
phone_number = data.get('phone_number')
|
@@ -112,9 +94,8 @@ def login():
|
|
112 |
if not name or not email or not phone_number:
|
113 |
return jsonify({'error': 'Missing required fields'}), 400
|
114 |
|
115 |
-
# Create a record in Salesforce
|
116 |
try:
|
117 |
-
|
118 |
return redirect("/menu") # Redirect to the menu page after successful login
|
119 |
except Exception as e:
|
120 |
return jsonify({'error': f'Failed to create record in Salesforce: {str(e)}'}), 500
|
@@ -146,53 +127,30 @@ def submit():
|
|
146 |
return jsonify({'error': str(e)}), 500
|
147 |
|
148 |
|
149 |
-
# Route for the menu page
|
150 |
@app.route("/menu", methods=["GET"])
|
151 |
def menu_page():
|
152 |
-
menu_items = get_menu_items() # Fetch menu items from Salesforce
|
153 |
-
menu_data = []
|
154 |
-
for item in menu_items:
|
155 |
-
menu_data.append({
|
156 |
-
"name": item['Name'],
|
157 |
-
"price": item['Price__c'],
|
158 |
-
"ingredients": item['Ingredients__c'],
|
159 |
-
"category": item['Category__c'],
|
160 |
-
})
|
161 |
-
|
162 |
-
# Render the menu page template and pass the menu data to it
|
163 |
return render_template("menu_page.html", menu_items=menu_data)
|
164 |
|
165 |
-
# Route for handling order
|
166 |
@app.route("/order", methods=["POST"])
|
167 |
def place_order():
|
168 |
-
# Getting the item name from the POST request (you can customize this to get more data)
|
169 |
item_name = request.json.get('item_name')
|
170 |
quantity = request.json.get('quantity')
|
171 |
-
|
172 |
-
# Logic to save the order details to Salesforce
|
173 |
-
order_data = {
|
174 |
-
"Item__c": item_name, # Assuming 'Item__c' is the field in your Order object
|
175 |
-
"Quantity__c": quantity, # Assuming 'Quantity__c' is a field for the quantity
|
176 |
-
# Add any other fields you need to capture, like customer, price, etc.
|
177 |
-
}
|
178 |
-
|
179 |
-
# Create the order record in Salesforce
|
180 |
sf.Order__c.create(order_data)
|
181 |
-
|
182 |
-
# Return a success response
|
183 |
return jsonify({"success": True, "message": f"Order for {item_name} placed successfully."})
|
184 |
|
185 |
-
# Route to handle the cart
|
186 |
@app.route("/cart", methods=["GET"])
|
187 |
def cart():
|
188 |
-
# Logic to fetch cart items (this could be from a session or database)
|
189 |
cart_items = [] # Placeholder for cart items
|
190 |
return render_template("cart_page.html", cart_items=cart_items)
|
191 |
|
192 |
-
# Route for the order summary page
|
193 |
@app.route("/order-summary", methods=["GET"])
|
194 |
def order_summary():
|
195 |
-
# Logic to fetch order summary (this could be from a session or database)
|
196 |
order_details = [] # Placeholder for order details
|
197 |
return render_template("order_summary.html", order_details=order_details)
|
198 |
|
@@ -246,7 +204,7 @@ def transcribe():
|
|
246 |
confirmation = f"Is this correct? Name: {name}, Email: {email}, Phone: {phone_number}"
|
247 |
generate_audio_prompt(confirmation, "confirmation.mp3")
|
248 |
|
249 |
-
# Simulate confirmation via user action
|
250 |
user_confirms = True # Assuming the user confirms, you can replace this with actual user input logic
|
251 |
|
252 |
if user_confirms:
|
@@ -261,7 +219,6 @@ def transcribe():
|
|
261 |
print(f"Error creating record in Salesforce: {salesforce_response['error']}")
|
262 |
return jsonify(salesforce_response), 500
|
263 |
|
264 |
-
# If creation was successful, return the details
|
265 |
return jsonify({"text": transcribed_text, "salesforce_record": salesforce_response})
|
266 |
|
267 |
except Exception as e:
|
|
|
21 |
config = AutoConfig.from_pretrained("openai/whisper-small")
|
22 |
config.update({"timeout": 60}) # Set timeout to 60 seconds
|
23 |
|
24 |
+
# Salesforce connection details
|
25 |
+
try:
|
26 |
+
print("Attempting to connect to Salesforce...")
|
27 |
+
sf = Salesforce(username='your_username', password='your_password', security_token='your_token')
|
28 |
+
print("Connected to Salesforce successfully!")
|
29 |
+
print("User Info:", sf.UserInfo) # Log the user info to verify the connection
|
30 |
+
except Exception as e:
|
31 |
+
print(f"Failed to connect to Salesforce: {str(e)}")
|
32 |
+
|
33 |
+
# Functions for Salesforce operations
|
34 |
+
def create_salesforce_record(sf, name, email, phone_number):
|
35 |
+
try:
|
36 |
+
customer_login = sf.Customer_Login__c.create({
|
37 |
+
'Name': name,
|
38 |
+
'Email__c': email,
|
39 |
+
'Phone_Number__c': phone_number
|
40 |
+
})
|
41 |
+
return customer_login
|
42 |
+
except Exception as e:
|
43 |
+
raise Exception(f"Failed to create record: {str(e)}")
|
44 |
+
|
45 |
+
def get_menu_items(sf):
|
46 |
+
query = "SELECT Name, Price__c, Ingredients__c, Category__c FROM Menu_Item__c"
|
47 |
+
result = sf.query(query)
|
48 |
+
return result['records']
|
49 |
+
|
50 |
+
# Voice-related functions
|
51 |
def generate_audio_prompt(text, filename):
|
52 |
try:
|
53 |
tts = gTTS(text)
|
|
|
58 |
time.sleep(5) # Wait for 5 seconds before retrying
|
59 |
generate_audio_prompt(text, filename)
|
60 |
|
61 |
+
# Utility functions
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
def convert_to_wav(input_path, output_path):
|
63 |
try:
|
64 |
audio = AudioSegment.from_file(input_path)
|
|
|
68 |
print(f"Error: {str(e)}")
|
69 |
raise Exception(f"Audio conversion failed: {str(e)}")
|
70 |
|
|
|
71 |
def is_silent_audio(audio_path):
|
72 |
audio = AudioSegment.from_wav(audio_path)
|
73 |
nonsilent_parts = detect_nonsilent(audio, min_silence_len=500, silence_thresh=audio.dBFS-16) # Reduced silence duration
|
74 |
print(f"Detected nonsilent parts: {nonsilent_parts}")
|
75 |
return len(nonsilent_parts) == 0 # If no speech detected
|
76 |
|
77 |
+
# Routes and Views
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
78 |
@app.route("/")
|
79 |
def index():
|
80 |
return render_template("index.html")
|
81 |
|
82 |
+
@app.route("/dashboard", methods=["GET"])
|
83 |
def dashboard():
|
|
|
84 |
return render_template("dashboard.html") # Render the dashboard template
|
85 |
|
|
|
|
|
86 |
@app.route('/login', methods=['POST'])
|
87 |
def login():
|
88 |
# Get data from voice bot (name, email, phone number)
|
89 |
data = request.json # Assuming voice bot sends JSON data
|
|
|
90 |
name = data.get('name')
|
91 |
email = data.get('email')
|
92 |
phone_number = data.get('phone_number')
|
|
|
94 |
if not name or not email or not phone_number:
|
95 |
return jsonify({'error': 'Missing required fields'}), 400
|
96 |
|
|
|
97 |
try:
|
98 |
+
customer_login = create_salesforce_record(sf, name, email, phone_number)
|
99 |
return redirect("/menu") # Redirect to the menu page after successful login
|
100 |
except Exception as e:
|
101 |
return jsonify({'error': f'Failed to create record in Salesforce: {str(e)}'}), 500
|
|
|
127 |
return jsonify({'error': str(e)}), 500
|
128 |
|
129 |
|
|
|
130 |
@app.route("/menu", methods=["GET"])
|
131 |
def menu_page():
|
132 |
+
menu_items = get_menu_items(sf) # Fetch menu items from Salesforce
|
133 |
+
menu_data = [{"name": item['Name'], "price": item['Price__c'], "ingredients": item['Ingredients__c'], "category": item['Category__c']} for item in menu_items]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
134 |
return render_template("menu_page.html", menu_items=menu_data)
|
135 |
|
136 |
+
# Route for handling order
|
137 |
@app.route("/order", methods=["POST"])
|
138 |
def place_order():
|
|
|
139 |
item_name = request.json.get('item_name')
|
140 |
quantity = request.json.get('quantity')
|
141 |
+
order_data = {"Item__c": item_name, "Quantity__c": quantity}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
142 |
sf.Order__c.create(order_data)
|
|
|
|
|
143 |
return jsonify({"success": True, "message": f"Order for {item_name} placed successfully."})
|
144 |
|
145 |
+
# Route to handle the cart
|
146 |
@app.route("/cart", methods=["GET"])
|
147 |
def cart():
|
|
|
148 |
cart_items = [] # Placeholder for cart items
|
149 |
return render_template("cart_page.html", cart_items=cart_items)
|
150 |
|
151 |
+
# Route for the order summary page
|
152 |
@app.route("/order-summary", methods=["GET"])
|
153 |
def order_summary():
|
|
|
154 |
order_details = [] # Placeholder for order details
|
155 |
return render_template("order_summary.html", order_details=order_details)
|
156 |
|
|
|
204 |
confirmation = f"Is this correct? Name: {name}, Email: {email}, Phone: {phone_number}"
|
205 |
generate_audio_prompt(confirmation, "confirmation.mp3")
|
206 |
|
207 |
+
# Simulate confirmation via user action
|
208 |
user_confirms = True # Assuming the user confirms, you can replace this with actual user input logic
|
209 |
|
210 |
if user_confirms:
|
|
|
219 |
print(f"Error creating record in Salesforce: {salesforce_response['error']}")
|
220 |
return jsonify(salesforce_response), 500
|
221 |
|
|
|
222 |
return jsonify({"text": transcribed_text, "salesforce_record": salesforce_response})
|
223 |
|
224 |
except Exception as e:
|