Spaces:
Build error
Build error
File size: 2,562 Bytes
d483661 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
import json
from whisper_stt import transcribe_audio_raw
from intent_recognition import get_intent_and_amount
from generate_response import generate_voice_response
from test_NLU import get_slots
DATABASE_PATH = "database.json"
def load_database():
with open(DATABASE_PATH, "r") as db_file:
return json.load(db_file)
def save_to_database(data):
with open(DATABASE_PATH, "w") as db_file:
json.dump(data, db_file, indent=4)
def handle_request(audio_file):
text = transcribe_audio_raw(audio_file)
intent_data = get_intent_and_amount(text)
intent=intent_data.get('intent')
intent=intent.replace("_", " ").title()
amount_data=intent_data.get('amount_data')
amount=amount_data.get('amount')
currency=amount_data.get('currency')
slots=get_slots(text)
slots['amount']=amount+' '+currency
if intent is not None:
response=f"You have requested for the task: Intent: {intent}, Project: {slots.get('project_name')}. Project ID: {slots.get('project_id')}. Amount: {slots.get('amount')}. Task ID: {slots.get('task_id')} and Status: {slots.get('status')}. Please Confirm by typing your response: Yes or No: "
generate_voice_response(response)
user_input=input("Please type your response: Yes or No: ")
if user_input.lower()=="yes":
# Prepare the data to save
request_data = {
"project": slots.get("project_name"),
"project_id": slots.get("project_id"),
"amount": amount,
"Intent": intent,
"task_id": slots.get("task_id"),
"status": slots.get("status"),
}
database = load_database()
database["requests"].append(request_data)
save_to_database(database)
generate_voice_response("Thank you for your response, Your request has been confirmed successfully.")
elif user_input.lower()=="no":
generate_voice_response("Thank you for your response, You have denied the confirmation request.")
else:
generate_voice_response("You have typed an invalid response.")
else:
response = "Sorry, I did not understand your request."
generate_voice_response(response)
return response
if __name__ == "__main__":
user_audio = "input_audio.wav"
# audio_file = open(user_audio, "rb")
print(handle_request(user_audio))
|