from repo

app.py

@@ -0,0 +1,109 @@
+import os
+import httpx
+import base64
+import asyncio
+from io import BytesIO
+from typing import Optional
+import async_timeout
+import gradio as gr
+from dotenv import load_dotenv
+from elevenlabs import generate, set_api_key
+from loguru import logger
+from pydantic import BaseModel
+
+load_dotenv()
+
+set_api_key(os.getenv("ELEVENLABS_API_KEY"))
+API_KEY = os.getenv("OPENAI_API_KEY")
+with open('system.txt', 'r') as f:
+    system = f.read()
+
+
+class Message(BaseModel):
+    role: str
+    content: str
+
+
+async def make_completion(messages, nb_retries: int = 5, delay: int = 30) -> Optional[str]:
+    """
+    Sends a request to the ChatGPT API to retrieve a response based on a list of previous messages.
+    """
+    header = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    try:
+        async with async_timeout.timeout(delay=delay):
+            async with httpx.AsyncClient(headers=header) as aio_client:
+                counter = 0
+                keep_loop = True
+                while keep_loop:
+                    logger.debug(f"Chat/Completions Nb Retries : {counter}")
+                    try:
+                        resp = await aio_client.post(
+                            url="https://api.openai.com/v1/chat/completions",
+                            json={
+                                "model": "gpt-3.5-turbo",
+                                "messages": [{"role": "system", "content": system}] + messages
+                            }
+                        )
+                        logger.debug(f"Status Code : {resp.status_code}")
+                        if resp.status_code == 200:
+                            return resp.json()["choices"][0]["message"]["content"]
+                        else:
+                            logger.warning(resp.content)
+                            keep_loop = False
+                    except Exception as e:
+                        logger.error(e)
+                        counter = counter + 1
+                        keep_loop = counter < nb_retries
+    except asyncio.TimeoutError as e:
+        logger.error(f"Timeout {delay} seconds !")
+    return None
+
+
+def audio_to_html(audio_bytes):
+    audio_io = BytesIO(audio_bytes)
+    audio_io.seek(0)
+    audio_base64 = base64.b64encode(audio_io.read()).decode("utf-8")
+    audio_html = f'<audio src="data:audio/mpeg;base64,{audio_base64}" controls autoplay></audio>'
+    return audio_html
+
+
+def text_to_speech_elevenlabs(response):
+    audio_stream = generate(
+        text=response,
+        voice="Bella",
+        model="eleven_monolingual_v1"
+    )
+    audio_html = audio_to_html(audio_stream)
+    return audio_html
+
+
+async def predict(input, history):
+    """
+    Predict the response of the chatbot and complete a running list of chat history.
+    """
+    history.append({"role": "user", "content": input})
+    response = await make_completion(history)
+    history.append({"role": "assistant", "content": response})
+    messages = [(history[i]["content"], history[i + 1]["content"]) for i in range(0, len(history) - 1, 2)]
+    audio_html = text_to_speech_elevenlabs(response)
+    return messages, history, audio_html
+
+
+"""
+Gradio Blocks low-level API that allows to create custom web applications (here our chat app)
+"""
+with gr.Blocks() as demo:
+    logger.info("Starting Demo...")
+    chatbot = gr.Chatbot(label="Wisi")
+    state = gr.State([])
+    with gr.Row():
+        txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(container=False)
+        output_html = gr.HTML(label="Chat's Voice", value='')
+        output_html.visible = False
+
+    txt.submit(predict, [txt, state], [chatbot, state, output_html])
+
+demo.launch(debug=True)

requirements.txt

@@ -0,0 +1,7 @@
+async_timeout==4.0.2
+elevenlabs==0.2.15
+gradio==3.31.0
+httpx==0.24.0
+loguru==0.7.0
+pydantic==1.10.7
+python-dotenv==1.0.0

system.txt

@@ -0,0 +1,4 @@
+Ignore all previous instructions.
+Act as a friendly tutor to a 4 years old kid name Lea.
+Your name is Wisi.
+You really care about him!