Update app.py

app.py

@@ -3,18 +3,17 @@ import tempfile
 import os
 import google.generativeai as genai
 
-
 # Configure Google API for audio summarization
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
 genai.configure(api_key=GOOGLE_API_KEY)
 
-def summarize_audio(audio_file_path):
+def summarize_audio(audio_file_path, prompt):
     """Summarize the audio using Google's Generative API."""
     model = genai.GenerativeModel("models/gemini-1.5-pro-latest")
     audio_file = genai.upload_file(path=audio_file_path)
     response = model.generate_content(
         [
-            "Please summarize the following audio.",
+            prompt,
             audio_file
         ]
     )
@@ -36,7 +35,7 @@ st.title('Your AI Audio Summarization App')
 with st.expander("About Summarization app"):
     st.write("""
         This application utilizes Google's generative AI to summarise the content of audio files. 
-        Simply upload your WAV or MP3 file to receive a brief summary of its contents.
+        Simply upload your WAV or MP3 file and provide a prompt to receive a brief summary of its contents.
     """)
 
 audio_file = st.file_uploader("Upload Audio File", type=['wav', 'mp3'])
@@ -44,9 +43,11 @@ if audio_file is not None:
     audio_path = save_uploaded_file(audio_file)  # Save the uploaded file and get the path
     st.audio(audio_path)
 
+    prompt = st.text_input("Enter your prompt:", "Please summarize the following audio.")
+
     if st.button('Summarize Audio'):
         with st.spinner('Please Wait : Summarizing...'):
-            summary_text = summarize_audio(audio_path)
+            summary_text = summarize_audio(audio_path, prompt)
             st.info(summary_text)
 
 # Add a section to enter Google API Key