Update main.py

main.py

@@ -1,58 +1,58 @@
-import os
-from dotenv import load_dotenv
-import openai
-from flask import Flask, request, jsonify, send_file
+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse, HTMLResponse
+from fastapi.templating import Jinja2Templates
+
 from transformers import pipeline
-from gtts import gTTS
+from openai import OpenAI
 
-app = Flask(__name__)
+app = FastAPI()
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-# Load environment variables from .env file
-load_dotenv()
+templates = Jinja2Templates(directory="templates")
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
-os.environ["HF_HOME"] = os.getenv("HF_HOME")
 pipe = pipeline(model="seeafricatz/kiaziboraasr")
-
-def transcribe(audio_path):
-    text = pipe(audio_path)["text"]
-    return text
-
-def generate_response(transcribed_text):
-    response = openai.Completion.create(
-        engine="davinci",
-        prompt=transcribed_text,
-        max_tokens=50
+client = OpenAI()
+
+@app.post("/transcribe")
+async def transcribe(request: Request):
+    form_data = await request.form()
+    audio_file = form_data["audio"]
+    text = pipe(audio_file)["text"]
+    return {"text": text}
+
+@app.post("/generate_response")
+async def generate_response(request: Request):
+    form_data = await request.form()
+    transcribed_text = form_data["text"]
+    response = client.chat.completions.create(
+        messages=[
+            {"role": "system", "content": "All your answers should be in swahili only, users undertands swahili only so here we start... Wewe ni mtaalamu wa viazi lishe na utajibu maswali yote kwa kiswahili tu!"},
+            {"role": "user", "content": "Mambo vipi?"},
+            {"role": "assistant", "content": "Salama je una swali lolote kuhusu viazi lishe?"},
+            {"role": "user", "content": "nini maana ya Viazi lishe?"},
+            {"role": "assistant", "content": "viazi lishe ni Viazi vitamu vyenye rangi ya karoti kwa ndani ambavyo vina vitamin A kwa wingi"},
+            {"role": "user", "content": "nini matumizi ya viazi lishe?"},
+            {"role": "assistant", "content": "viazi lishe vinaweza kutengenezea chakula kama Keki, Maandazi, Kalimati na tambi: Ukisaga unga wa viazi lishe, unaweza kutumika kupika vyakula ambavyo huwa watu hutumia unga wa ngano kupika, unga wa viazi lishe una virutubisho vingi zaidi kuliko unga wa ngano na ukitumika kupikia vyakula tajwa hapo juu watumiaji watakuwa wanakula vyakula vyenye virutubisho Zaidi."},
+            {"role": "user", "content": transcribed_text},
+        ],
+        model="gpt-4"
     )
-    return response.choices[0].text
-
-def inference(text):
-    tts = gTTS(text, lang='sw')
-    output_file = "tts_output.mp3"
-    tts.save(output_file)
-    return output_file
-
-@app.route('/process_audio', methods=['POST'])
-def process_audio():
-    if 'audio' not in request.files:
-        return jsonify({'error': 'No audio file provided'}), 400
-
-    audio_file = request.files['audio']
-    audio_path = "temp_audio.wav"
-    audio_file.save(audio_path)
-
-    transcribed_text = transcribe(audio_path)
-    response_text = generate_response(transcribed_text)
-    output_file = inference(response_text)
-
-    return jsonify({
-        'response_text': response_text,
-        'response_audio_url': f'/audio/{output_file}'
-    })
-
-@app.route('/audio/<path:filename>')
-def audio(filename):
-    return send_file(filename, as_attachment=True)
-
-if __name__ == '__main__':
-    app.run()
+    return {"response": response.choices[0].message.content}
+
+@app.post("/inference")
+async def inference(request: Request):
+    form_data = await request.form()
+    text = form_data["text"]
+    response = client.audio.speech.create(
+        model="tts-1",
+        voice="nova",
+        input=text
+    )
+    output_file = "static/tts_output.mp3"
+    response.stream_to_file(output_file)
+    return {"audio_url": "/static/tts_output.mp3"}
+
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})