Update app.py

app.py

@@ -4,24 +4,81 @@ import gradio as gr
 from PIL import Image
 import os
 
+os.system("pip install openai")
+import openai
+
+api_key = os.environ.get('api_key')
+openai.api_key = api_key
+
+whisper = gr.Interface.load(name="spaces/sanchit-gandhi/whisper-large-v2")
+
 from share_btn import community_icon_html, loading_icon_html, share_js
 
 token = os.environ.get('HF_TOKEN')
-whisper_to_gpt = gr.Blocks.load(name="spaces/fffiloni/whisper-to-chatGPT")
+
 tts = gr.Interface.load(name="spaces/Flux9665/IMS-Toucan")
 talking_face = gr.Blocks.load(name="spaces/fffiloni/one-shot-talking-face", api_key=token)
 
 def infer(audio):
-    whisper_to_gpt_response = whisper_to_gpt(audio, "translate", fn_index=0)
-    #print(gpt_response)
-    audio_response = tts(whisper_to_gpt_response[1], "English Text", "English Accent", "English Speaker's Voice", fn_index=0)
-    #image = Image.open(r"wise_woman_portrait.png")
+    
+    whisper_result = whisper(audio, None, "translate", fn_index=0)
+    
+    gpt_response = try_api(whisper_result)
+    
+    audio_response = tts(gpt_response[0], "English Text", "English Accent", "English Speaker's Voice", fn_index=0)
+    
     portrait_link = talking_face("wise_woman_portrait.png", audio_response, fn_index=0)
     
-    #portrait_response = requests.get(portrait_link, headers={'Authorization': 'Bearer ' + token})
-    #print(portrait_response.text)
+    return whisper_result, portrait_link, gr.update(visible=True)
+
+def try_api(message):
+    try:
+        response = call_api(message)
+        return response, "no error"
+    except openai.error.Timeout as e:
+        #Handle timeout error, e.g. retry or log
+        print(f"OpenAI API request timed out: {e}")
+        return "oups", f"OpenAI API request timed out: {e}"
+    except openai.error.APIError as e:
+        #Handle API error, e.g. retry or log
+        print(f"OpenAI API returned an API Error: {e}")
+        return "oups", f"OpenAI API returned an API Error: {e}"
+    except openai.error.APIConnectionError as e:
+        #Handle connection error, e.g. check network or log
+        print(f"OpenAI API request failed to connect: {e}")
+        return "oups", f"OpenAI API request failed to connect: {e}"
+    except openai.error.InvalidRequestError as e:
+        #Handle invalid request error, e.g. validate parameters or log
+        print(f"OpenAI API request was invalid: {e}")
+        return "oups", f"OpenAI API request was invalid: {e}"
+    except openai.error.AuthenticationError as e:
+        #Handle authentication error, e.g. check credentials or log
+        print(f"OpenAI API request was not authorized: {e}")
+        return "oups", f"OpenAI API request was not authorized: {e}"
+    except openai.error.PermissionError as e:
+        #Handle permission error, e.g. check scope or log
+        print(f"OpenAI API request was not permitted: {e}")
+        return "oups", f"OpenAI API request was not permitted: {e}"
+    except openai.error.RateLimitError as e:
+        #Handle rate limit error, e.g. wait or log
+        print(f"OpenAI API request exceeded rate limit: {e}")
+        return "oups", f"OpenAI API request exceeded rate limit: {e}"
+
+def call_api(message):
+          
+    print("starting open ai")
     
-    return whisper_to_gpt_response[0], portrait_link, gr.update(visible=True)
+    response = openai.Completion.create(
+        model="text-davinci-003",
+        prompt=message,
+        temperature=0.5,
+        max_tokens=2048,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0.6
+    )
+
+    return str(response.choices[0].text).split("\n",2)[2]
 
 title = """
     <div style="text-align: center; max-width: 500px; margin: 0 auto;">