diff --git a/py/simple_chat.py b/py/simple_chat.py index 0c3165f..879e1f1 100644 --- a/py/simple_chat.py +++ b/py/simple_chat.py @@ -1,6 +1,5 @@ from api import API - class CLIChat: @staticmethod def chat(): diff --git a/py/static/index.html b/py/templates/index.html similarity index 88% rename from py/static/index.html rename to py/templates/index.html index cef6982..3425d0a 100644 --- a/py/static/index.html +++ b/py/templates/index.html @@ -71,16 +71,19 @@
-
User: What is the weather today?
-
AI: It's sunny with a slight breeze.
-
User: Great! Thank you!
-
AI: You're welcome!
+ {% for message in messages %} + {% if message.startswith('User:') %} +
{{ message }}
+ {% else %} +
{{ message }}
+ {% endif %} + {% endfor %}
- +
diff --git a/py/voice_recognition.py b/py/voice_recognition.py new file mode 100644 index 0000000..0ee41d0 --- /dev/null +++ b/py/voice_recognition.py @@ -0,0 +1,33 @@ +import speech_recognition as sr + + +class Voice: + @staticmethod + def listen(): + recognizer = sr.Recognizer() + + try: + recognizer = sr.Recognizer() + with sr.Microphone() as source: + print("Adjusting for ambient noise...") + recognizer.adjust_for_ambient_noise(source) + print("Listening...") + audio_data = recognizer.listen(source) + print("Audio captured") + try: + text = recognizer.recognize_sphinx(audio_data) # Using Google Web Speech API + except sr.UnknownValueError: + text = "ERROR" + except sr.RequestError as e: + text = "ERROR" + + except sr.RequestError as e: + text = "ERROR" + except sr.UnknownValueError: + text = "ERROR" + except Exception as e: + text = "ERROR" + return text + print(listen()) + + diff --git a/py/web_flask.py b/py/web_flask.py index 35223f0..0f160e8 100644 --- a/py/web_flask.py +++ b/py/web_flask.py @@ -1,20 +1,29 @@ -from flask import Flask, request +from flask import Flask, request, render_template +from api import API APP = Flask(__name__) +api = API() +messages = [] @APP.route('/', methods=['GET', 'POST']) def index(): + global messages + if request.method == 'POST': - name = request.form['name'] option = request.form['option'] - if option == "voice": - print(name) - elif option == "chat": - print(name) + user_message = request.form['user_message'] - return APP.send_static_file('index.html') + if option == "voice": + messages.append(f"User: {user_message}") + elif option == "chat": + messages.append(f"User: {user_message}") + + ai_response = "AI: " + api.send_message(user_message, 1) + messages.append(ai_response) + + return render_template('index.html', messages=messages) if __name__ == '__main__':