Update app.py

app.py

@@ -1,19 +1,52 @@
+import os
+import io
 import streamlit as st
 from youtube_transcript_api import YouTubeTranscriptApi
 import re
+import logging
+import warnings
+from pydub import AudioSegment
+import tempfile
 from dotenv import load_dotenv
-import os
-from groq import Groq
+from openai import OpenAI
 
+# Load environment variables
+load_dotenv()
 
+# Initialize the OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# Load the API key from .env file
-load_dotenv()
-api_key = os.getenv('GROQ_API_KEY')
-client = Groq(api_key=api_key)
+# Set up logging
+logging.basicConfig(level=logging.INFO)
 
-if not api_key:
-    raise ValueError("API key is not set. Please check your .env file and ensure GROQ_API_KEY is set.")
+def convert_to_supported_format(file):
+    audio = AudioSegment.from_file(file)
+    buffer = io.BytesIO()
+    audio.export(buffer, format="wav")
+    buffer.seek(0)
+    return buffer
+
+def transcribe_audio(file):
+    logging.info("Transcribing audio file")
+    file = convert_to_supported_format(file)
+    logging.info("Converted file to WAV format")
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
+        temp_file.write(file.getvalue())
+        temp_file_path = temp_file.name
+
+    try:
+        with open(temp_file_path, "rb") as audio_file:
+            transcript = client.audio.transcriptions.create(
+                model="whisper-1", 
+                file=audio_file
+            )
+        return transcript.text
+    except Exception as e:
+        logging.error(f"Error in transcription: {str(e)}")
+        return f"Error in transcription: {str(e)}"
+    finally:
+        os.remove(temp_file_path)
 
 def get_transcript(url):
     try:
@@ -31,32 +64,39 @@ def get_transcript(url):
 def summarize_text(text):
     try:
         response = client.chat.completions.create(
-            model="mixtral-8x7b-32768",
+            model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": f"Please provide a concise summary of the following text:\n\n{text}"}
             ],
             max_tokens=150
         )
-
-        if response.choices and response.choices[0].message:
-            summary = response.choices[0].message.content.strip()
-            return summary
-        else:
-            return "No summary available."
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"Error in summarizing text: {str(e)}"
 
-def handle_summary(youtube_url):
-    transcript = get_transcript(youtube_url)
-    if "Error" in transcript:
-        return transcript
-    summary = summarize_text(transcript)
+def handle_summary(transcript_text):
+    if "Error" in transcript_text:
+        return transcript_text
+    summary = summarize_text(transcript_text)
     return summary
 
 st.title("YouTube Summary Generator")
-youtube_url = st.text_input("YouTube URL", placeholder="Enter YouTube URL here...")
 
-if st.button("Generate Summary"):
-    summary = handle_summary(youtube_url)
-    st.text_area("Summary", summary, height=200)
+option = st.selectbox("Choose input type", ("YouTube URL", "Upload audio/video file"))
+
+if option == "YouTube URL":
+    youtube_url = st.text_input("YouTube URL", placeholder="Enter YouTube URL here...")
+    if st.button("Generate Summary"):
+        transcript = get_transcript(youtube_url)
+        summary = handle_summary(transcript)
+        st.text_area("Summary", summary, height=200)
+
+elif option == "Upload audio/video file":
+    uploaded_file = st.file_uploader("Choose an audio or video file", type=["mp3", "wav", "mp4", "mov"])
+    if uploaded_file is not None:
+        if st.button("Generate Summary"):
+            with st.spinner('Transcribing audio...'):
+                transcript_text = transcribe_audio(uploaded_file)
+            summary = handle_summary(transcript_text)
+            st.text_area("Summary", summary, height=200)