import streamlit as st
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, prompt):
    """Summarize and extract 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(
        [
            prompt,
            audio_file
        ]
    )
    return response.text

def save_uploaded_file(uploaded_file):
    """Save uploaded file to a temporary file and return the path."""
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix='.' + uploaded_file.name.split('.')[-1]) as tmp_file:
            tmp_file.write(uploaded_file.getvalue())
            return tmp_file.name
    except Exception as e:
        st.error(f"Error handling uploaded file: {e}")
        return None

# Streamlit app interface
st.title('SoundScape Intelligence: Unveiling Insights from Audio')

with st.expander("About SoundScape Intelligence app"):
    st.write("""
        This application utilizes Google's generative AI to summarize and extract key points from audio files . 
        Simply upload your WAV or MP3 file and provide a prompt to receive insight from the files.
    """)

audio_file = st.file_uploader("Upload Audio File", type=['wav', 'mp3'])
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('Extract Insight from  Audio'):
        with st.spinner('Please Wait : Extracting Information...'):
            summary_text = summarize_audio(audio_path, prompt)
            st.info(summary_text)

# Add a section to enter Google API Key
st.sidebar.header("Google API Key")
new_key = st.sidebar.text_input("Enter your Google API Key:", value=GOOGLE_API_KEY)
if new_key:
    os.environ["GOOGLE_API_KEY"] = new_key
    GOOGLE_API_KEY = new_key
    genai.configure(api_key=GOOGLE_API_KEY)