jf

app.py

@@ -24,18 +24,18 @@ def summarize_transcript(transcript):
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary
 
-def summarize_video_url(video_url):
-    transcript = get_transcript(video_url)
-    if not transcript:
-        return "Error fetching transcript."
-    else:
-        summary = summarize_transcript(transcript)
-        return summary
-
-inputs = gr.inputs.Textbox(lines=5, label="Enter YouTube Video URL")
-output = gr.outputs.Textbox(label="Summary")
+def summarize_video_transcript(url):
+    transcript = get_transcript(url)
+    if not transcript or transcript.startswith("Error"):
+        return transcript
+    summary = summarize_transcript(transcript)
+    return summary
 
-title = "YouTube Video Transcription Summarizer"
-description = "Enter a YouTube Video URL to get a summary of its transcript."
+iface = gr.Interface(fn=summarize_video_transcript,
+                     inputs="text",
+                     outputs="text",
+                     title="YouTube Video Transcription Summarizer",
+                     description="Enter YouTube Video URL to get the transcript summary")
 
-gr.Interface(fn=summarize_video_url, inputs=inputs, outputs=output, title=title, description=description).launch()
+if __name__ == "__main__":
+    iface.launch()