from flask import Flask, render_template, request, jsonify, send_file
import requests
from dotenv import load_dotenv
import os

# import namespaces
from azure.core.credentials import AzureKeyCredential
from azure.ai.language.questionanswering import QuestionAnsweringClient
from azure.cognitiveservices.speech import SpeechConfig, SpeechSynthesizer, AudioConfig
from azure.cognitiveservices.speech.audio import AudioOutputConfig

app = Flask(__name__)

# Azure Bot Service configuration
AZURE_BOT_ENDPOINT = "https://iti109-sectionb.cognitiveservices.azure.com/"
AZURE_BOT_KEY = "2ou0CMAjUutj0D4In8U8AkxEIXtCrvYFOBMhqSW4rZ7x6yZ033GdJQQJ99ALACqBBLyXJ3w3AAAaACOGtVJj"

# Get Configuration Settings
load_dotenv()
ai_endpoint = os.getenv('AI_SERVICE_ENDPOINT')
ai_key = os.getenv('AI_SERVICE_KEY')
ai_project_name = os.getenv('QA_PROJECT_NAME')
ai_deployment_name = os.getenv('QA_DEPLOYMENT_NAME')
speech_key = os.getenv('SPEECH_KEY')
speech_region = os.getenv('SPEECH_REGION')

# Create client using endpoint and key
credential = AzureKeyCredential(ai_key)
ai_client = QuestionAnsweringClient(endpoint=ai_endpoint, credential=credential)

@app.route('/')
def home():
    return render_template('index.html')  # HTML file for the web interface

@app.route('/ask', methods=['POST'])
def ask_bot():
    user_question = request.json.get("question", "")
    
    if not user_question:
        return jsonify({"error": "No question provided"}), 400

    try:
        response = ai_client.get_answers(question=user_question,
                                         project_name=ai_project_name,
                                         deployment_name=ai_deployment_name)                                           
           
        bot_response = response.answers[0].answer if response.answers else "No response from bot"
        
        # Text-to-Speech
        speech_config = SpeechConfig(subscription=speech_key, region=speech_region)
        audio_config = AudioConfig(filename="response.wav")
        synthesizer = SpeechSynthesizer(speech_config=speech_config, audio_config=audio_config)
        synthesizer.speak_text(bot_response)

        return jsonify({"answer": bot_response, "audio": "/response.wav"})       
    except requests.exceptions.RequestException as e:
        return jsonify({"error": str(e)}), 500

@app.route('/response.wav')
def get_audio():
    return send_file("response.wav", mimetype="audio/wav")

if __name__ == '__main__':
    app.run(debug=True)