Update app.py

app.py

@@ -1,17 +1,14 @@
 import streamlit as st
 import requests
+import io
 
-# Title of the app
-st.title("Whisper API Transcription Service")
-st.markdown("Upload an audio file (mp3, wav, or flac) and get transcription results.")
+# Hugging Face API setup
+API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3-turbo"
 
 # Prompt the user for their Hugging Face API key
 my_key = st.text_input('Enter your Hugging Face API Key', type='password')
 
-# API URL for Whisper model
-API_URL = "https://api-inference.huggingface.co/models/openai/whisper-large-v3-turbo"
-
-# Set up the headers with the provided API key
+# Set up headers with the provided API key
 headers = {"Authorization": f"Bearer {my_key}"}
 
 # Function to send the file to the API and get the transcription result
@@ -32,6 +29,14 @@ if my_key:  # Proceed only if the API key is provided
         for uploaded_file in uploaded_files:
             st.write(f"Processing file: {uploaded_file.name}")
             
+            # Validate file type and check if it's in the correct format
+            file_type = uploaded_file.type
+            st.write(f"File type: {file_type}")
+            
+            if file_type not in ["audio/mpeg", "audio/wav", "audio/flac"]:
+                st.write(f"Unsupported file type: {file_type}. Please upload an MP3, WAV, or FLAC file.")
+                continue
+
             # Send the file to the Hugging Face API
             output = query(uploaded_file)
             
@@ -47,4 +52,3 @@ if my_key:  # Proceed only if the API key is provided
         st.write("Please upload an audio file to transcribe.")
 else:
     st.write("Please enter your Hugging Face API key to proceed.")
-