main.py

import os
from flask import Flask, request, jsonify, url_for, render_template
import speech_recognition as sr
import logging
import io
from google.oauth2 import service_account
from gtts import gTTS
import uuid
from google.cloud import dialogflow

app = Flask(__name__)
logging.basicConfig(level=logging.DEBUG)

@app.route('/stt', methods=['POST'])
def transcribe_audio():
    app.logger.debug("Received a request with the following files: %s", request.files)
    if 'file' not in request.files:
        app.logger.error("No file part in the request")
        return jsonify({"error": "No file part"}), 400

    file = request.files['file']
    if file.filename == '':
        app.logger.error("No file selected")
        return jsonify({"error": "No selected file"}), 400

    if file:
        recognizer = sr.Recognizer()
        try:
            audio_data = sr.AudioFile(io.BytesIO(file.read()))
            with audio_data as source:
                audio = recognizer.record(source)
            text = recognizer.recognize_google(audio, language='ko-KR')
            return jsonify({"transcript": text})
        except sr.UnknownValueError:
            app.logger.error("Speech recognition could not understand audio")
            return jsonify({"error": "Google Speech Recognition could not understand audio"}), 400
        except sr.RequestError as e:
            app.logger.error("Could not request results from Google Speech Recognition service: %s", e)
            return jsonify({"error": f"Could not request results from Google Speech Recognition service; {e}"}), 500
        except Exception as e:
            app.logger.exception("An unexpected error occurred")
            return jsonify({"error": f"An unexpected error occurred: {e}"}), 500

@app.route('/tts', methods=['POST'])
def text_to_speech():
    data = request.json
    if 'text' not in data:
        return jsonify({"error": "No text provided"}), 400

    text = data['text']
    try:
        tts = gTTS(text=text, lang='ko')
        filename = f"output_{uuid.uuid4()}.mp3"
        tts.save(filename)

        if not os.path.exists(filename):
            return jsonify({"error": "File not found after saving"}), 500

        # Return the full path of the file
        return jsonify({"filename": os.path.abspath(filename)})
    except Exception as e:
        return jsonify({"error": str(e)}), 500

@app.route('/send-message', methods=['POST'])
def send_message():
    try:
        # Load credentials explicitly
        credentials = service_account.Credentials.from_service_account_file(
            r'E:\kioskdialouge-2c40828c055f.json')

        # Dialogflow 세션 클라이언트 설정
        session_client = dialogflow.SessionsClient(credentials=credentials)
    except Exception as e:
        app.logger.error("Failed to create Dialogflow session client: %s", e)
        return jsonify({"error": "Failed to connect to Dialogflow"}), 500

    try:
        # 요청에서 메시지 추출
        request_data = request.json
        message = request_data['message']
        session_id = request_data['session_id']

        # Dialogflow 세션 ID와 프로젝트 ID 설정
        project_id = 'kioskdialouge'
        session = session_client.session_path(project_id, session_id)

        # 텍스트 입력 설정
        text_input = dialogflow.TextInput(text=message, language_code="ko-KR")
        query_input = dialogflow.QueryInput(text=text_input)

        # Dialogflow에 요청 보내기
        response = session_client.detect_intent(session=session, query_input=query_input)

        if response.query_result.fulfillment_text:
            response_text = response.query_result.fulfillment_text
        else:
            response_text = "Dialogflow did not return a response."

        # 현재 단계 확인 및 업데이트
        if 'step' not in session:
            session['step'] = 1

        redirect_url = None
        if response_text != "이해하지 못했어요. 다시 한 번 말씀해주시겠어요?":
            session['step'] += 1

        if session['step'] == 1:
            redirect_url = url_for('sandwich')
        elif session['step'] == 2:
            redirect_url = url_for('bread')
        elif session['step'] == 3:
            redirect_url = url_for('vege')
        elif session['step'] == 4:
            redirect_url = url_for('side')
        elif session['step'] == 5:
            redirect_url = url_for('sc')
        elif session['step'] == 6:
            redirect_url = url_for('add')
        elif session['step'] == 7:
            redirect_url = url_for('check')
        elif session['step'] == 8:
            redirect_url = url_for('payment')
        else:
            session['step'] = 1  # 다시 처음으로


        return jsonify({'reply': response_text, 'session_id': session_id, 'redirect': redirect_url})


    except KeyError:
        app.logger.error("Invalid request: 'message' or 'session_id' field is missing")
        return jsonify({"error": "Invalid request: 'message' or 'session_id' field is missing"}), 400
    except Exception as e:
        app.logger.error("Failed to communicate with Dialogflow: %s", e)
        return jsonify({"error": f"Failed to communicate with Dialogflow: {e}"}), 500

@app.route("/add")
def add():
    return render_template('add.html')

@app.route("/bread")
def bread():
    return render_template('bread.html')

@app.route("/check")
def check():
    return render_template('check.html')

@app.route("/credit")
def credit():
    return render_template('credit.html')

@app.route("/dialog")
def dialog():
    return render_template('dialog.html')

@app.route("/lastpage")
def lastpage():
    return render_template('lastpage.html')

@app.route("/mainpage")
def mainpage():
    return render_template('mainpage.html')

@app.route("/payment")
def payment():
    return render_template('payment.html')

@app.route("/sandwich")
def sandwich():
    return render_template('sandwich.html')

@app.route("/sc")
def sc():
    return render_template('sc.html')

@app.route("/side")
def side():
    return render_template('side.html')

@app.route("/vege")
def vege():
    return render_template('vege.html')


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