Update main.py

main.py

@@ -1,14 +1,17 @@
 import os
+from dotenv import load_dotenv
 import openai
 from flask import Flask, request, jsonify, send_file
 from transformers import pipeline
-from openai import OpenAI
 from gtts import gTTS
 
 app = Flask(__name__)
-client = OpenAI()
 
-openai.api_key = "sk-proj-zufatzHdsNmcalxgJUG7T3BlbkFJJCw6YnyYqkJhS4xkvrm"
+# Load environment variables from .env file
+load_dotenv()
+
+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):
@@ -16,14 +19,12 @@ def transcribe(audio_path):
     return text
 
 def generate_response(transcribed_text):
-    response = client.chat_completions.create(
-        messages=[
-            {"role": "system", "content": "All your answers should be in Swahili only..."},
-            {"role": "user", "content": transcribed_text},
-        ],
-        model="gpt-4"
+    response = openai.Completion.create(
+        engine="davinci",
+        prompt=transcribed_text,
+        max_tokens=50
     )
-    return response.choices[0].message.content
+    return response.choices[0].text
 
 def inference(text):
     tts = gTTS(text, lang='sw')