Update app.py

app.py

@@ -6,6 +6,18 @@ import os
 from datetime import datetime
 from audio_recorder_streamlit import audio_recorder
 
+API_URL = f'https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud'
+headers = {
+	"Authorization": "Bearer XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
+	"Content-Type": "audio/wav"
+}
+
+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 generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
@@ -24,34 +36,10 @@ def save_and_play_audio(audio_recorder):
         return filename
 
 # 9B. Speech transcription to file output - OPENAI Whisper
-def transcribe_audio(file_path):
-    key=os.getenv('IE_KEY')
-    headers = {
-	"Authorization": f"Bearer {key}",
-	"Content-Type": "audio/wav"
-    }
-    with open(file_path, 'rb') as f:
-        data = {'file': f}
-        API_URL = "https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud"
-        response = requests.post(API_URL, headers=headers, data=data) 
-        st.write(response)
-
-    if response.status_code == 200:
-        st.write('Reasoning with your transcription..')
-        try:
-            transcript=response.json().get('text')
-        except:
-            transcript=response
+def transcribe_audio(filename):
+    output = query(filename)
+    return output
 
-        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")
@@ -61,8 +49,7 @@ def main():
     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.write(transcription)
 
 if __name__ == "__main__":
     main()