Update app.py

app.py

@@ -24,7 +24,7 @@ import pathlib
 import pandas
 import numpy
 import textblob
-import gradio as gr
+import streamlit as st
 from youtube_transcript_api import YouTubeTranscriptApi
 import google.generativeai as genai
 import requests
@@ -64,7 +64,7 @@ def install_missing_packages():
 
 install_missing_packages()
 import os
-import gradio as gr
+import streamlit as st
 from youtube_transcript_api import YouTubeTranscriptApi
 from dotenv import load_dotenv
 import google.generativeai as genai
@@ -73,11 +73,8 @@ import requests
 from transformers import pipeline
 from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
 from pytube import YouTube
-import gradio as gr
-from transformers import pipeline
-from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
-from pytube import YouTube
-import gradio as gr
+
+
 
 hf_token = os.getenv("HF_TOKEN")
 if hf_token:
@@ -85,6 +82,12 @@ if hf_token:
 else:
     raise ValueError("HF_TOKEN environment variable not set.")
     
+import os
+import streamlit as st
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
+from pytube import YouTube
+from transformers import pipeline
+
 # Function to extract transcript from a YouTube video
 def extract_transcript(youtube_video_url):
     try:
@@ -127,38 +130,32 @@ def summarize_transcript(transcript_text):
             if chunk.strip():
                 out = summarizer(chunk, max_length=130, min_length=30, do_sample=False)
                 summarized_text.append(out[0]['summary_text'])
-        
+
         return " ".join(summarized_text)
     except Exception as e:
         return f"Error during summarization: {str(e)}"
 
-# Gradio Interface function
-def gradio_interface(youtube_url):
-    transcript = extract_transcript(youtube_url)
-    if transcript.startswith("Error"):
-        return transcript, "Summarization cannot be performed due to transcript error."
-    
-    summarized_text = summarize_transcript(transcript)
-    return transcript, summarized_text
-
-# Gradio Interface Design
-with gr.Blocks() as demo:
-    gr.Markdown("# YouTube Video Summarizer")
-    with gr.Row():
-        youtube_url_input = gr.Textbox(
-            label="YouTube Video URL", 
-            placeholder="Enter the YouTube video link here"
-        )
-    summarize_button = gr.Button("Summarize Transcript")
-    with gr.Row():
-        transcript_output = gr.Textbox(label="Original Transcript", lines=10, interactive=False)
-        summary_output = gr.Textbox(label="Summarized Text", lines=10, interactive=False)
-    summarize_button.click(
-        fn=gradio_interface, 
-        inputs=[youtube_url_input], 
-        outputs=[transcript_output, summary_output]
-    )
-
-# Launch Gradio App
-if __name__ == "__main__":
-    demo.launch()
+# Streamlit App Design
+st.title("YouTube Video Summarizer")
+
+# Input field for YouTube URL
+youtube_url = st.text_input("Enter YouTube Video URL", "")
+
+if st.button("Summarize Transcript"):
+    if youtube_url:
+        with st.spinner("Extracting transcript..."):
+            transcript = extract_transcript(youtube_url)
+
+        if transcript.startswith("Error"):
+            st.error(transcript)
+        else:
+            st.subheader("Original Transcript")
+            st.text_area("Transcript", transcript, height=200)
+
+            with st.spinner("Summarizing transcript..."):
+                summarized_text = summarize_transcript(transcript)
+
+            st.subheader("Summarized Text")
+            st.text_area("Summary", summarized_text, height=200)
+    else:
+        st.error("Please enter a valid YouTube URL.")