| import gradio as gr | |
| from tool import YouTubeTranscriptExtractor, TranscriptSummarizer | |
| youtube_tool = YouTubeTranscriptExtractor() | |
| summarizer_tool = TranscriptSummarizer() | |
| def process_youtube_video(video_url): | |
| transcript = youtube_tool.forward(video_url=video_url) | |
| summary_and_blog = summarizer_tool.forward(transcript=transcript) | |
| try: | |
| summary, image_url = summary_and_blog.split("\n\nImage URL: ") | |
| except ValueError: | |
| summary = summary_and_blog | |
| image_url = None | |
| return transcript, summary, image_url | |
| iface = gr.Interface( | |
| fn=process_youtube_video, | |
| inputs="text", | |
| outputs=[ | |
| gr.Textbox(label="Transcript"), | |
| gr.Textbox(label="Summary and Blog Content"), | |
| gr.Image(label="Generated Image", image_mode="RGBA") | |
| ], | |
| title="YouTube Transcript Summarizer and Blog Content Generator", | |
| description="Enter a YouTube video URL to extract the transcript, summarize it, and generate blog content with an image." | |
| ) | |
| iface.launch() | |