Update app.py

app.py

@@ -2,41 +2,46 @@ import os
 import subprocess
 import openai
 import gradio as gr
+import requests
+from gtts import gTTS
 
+API_URL = "https://api-inference.huggingface.co/models/lyimo/whisper-small-sw2"
+headers = {"Authorization": "Bearer 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"}
 
-import os
+openai.api_key = "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"
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+def query(filename):
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.post(API_URL, headers=headers, data=data)
+    return response.json()
 
 
 def transcribe(audio):
-   with open(audio, "rb") as audio_file:
-       transcript = openai.Audio.transcribe("whisper-1", audio_file)
-   return transcript["text"]
-
+    output = query(audio)
+    return output["text"]
 
 def generate_response(transcribed_text):
-   response = openai.ChatCompletion.create(
-       model="gpt-3.5-turbo",
-       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},
-       ]
-   )
-   return response['choices'][0]['message']['content']
-
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "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},
+        ]
+    )
+    return response['choices'][0]['message']['content']
 
 def inference(text):
     output_file = "tts_output.wav"