Update app.py

app.py

@@ -4,13 +4,6 @@ import streamlit as st
 from datetime import datetime
 from audio_recorder_streamlit import audio_recorder
 
-API_URL = "https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud"
-key = 'test-public-anonymous-T4-Whisper-Small-En'
-headers = {
-    "Authorization": "Bearer {key}",
-    "Content-Type": "audio/wav"
-}
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
@@ -18,39 +11,56 @@ def generate_filename(prompt, file_type):
     safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
-def query(filename):
-    with open(filename, "rb") as f:
-        data = f.read()
-    response = requests.post(API_URL, headers=headers, data=data)
-    return response.json()
-
+# 10. Audio recorder to Wav file:
 def save_and_play_audio(audio_recorder):
-    audio_bytes = audio_recorder.get_audio()
+    audio_bytes = audio_recorder()
     if audio_bytes:
         filename = generate_filename("Recording", "wav")
         with open(filename, 'wb') as f:
             f.write(audio_bytes)
         st.audio(audio_bytes, format="audio/wav")
         return filename
-    return None
 
+# 9B. Speech transcription to file output - OPENAI Whisper
 def transcribe_audio(file_path):
-    API_URL = "https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud"
+    key=os.getenv('IE_KEY')
     headers = {
-        "Authorization": f"Bearer {openai_key}",
+	"Authorization": f"Bearer {key}",
+	"Content-Type": "audio/wav"
     }
     with open(file_path, 'rb') as f:
         data = {'file': f}
-        response = requests.post(API_URL, headers=headers, files=data)
+        API_URL = "https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud"
+        response = requests.post(API_URL, headers=headers, data=data) 
+
     if response.status_code == 200:
-        st.write(response.json())
+        st.write('Reasoning with your transcription..')
+        try:
+            transcript=response.json().get('text')
+        except:
+            transcript=response
+
+        st.write(transcript)
+        #gptResponse = chat_with_model(transcript, systemPrompt=user_prompt_System, model=MODELCHOICE) # send transcript to ChatGPT - prompts, systemPrompt=SYSTEM_PROMPT, model="Gpt-4-32k"
+        #filename = generate_filename(transcript, choice)
+        #create_file(filename, transcript, gptResponse) # write output file
+        #return gptResponse
+    else:
+        #st.write(response.json())
+        #st.error("Error in API call.")
+        return None
+
+def main():
+    st.title("Speech to Text")
+    st.write("Record your speech and get the text.")
+
+    # Audio, transcribe, GPT:
+    filename = save_and_play_audio(audio_recorder)
+    if filename is not None:
+        transcription = transcribe_audio(filename)
+        #st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+        filename = None
 
-st.title("Speech to Text")
-st.write("Record your speech and get the text.")
+if __name__ == "__main__":
+    main()
 
-# Audio, transcribe, GPT:
-filename = save_and_play_audio(audio_recorder)
-if filename is not None:
-    transcription = transcribe_audio(filename)
-    #st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-    filename = None