File size: 5,588 Bytes
56ac5db 41a98e7 583462f 56ac5db 41a98e7 b984651 51fd165 cc4b72f 41a98e7 cc4b72f 41a98e7 56ac5db 41a98e7 56ac5db 41a98e7 cc4b72f 41a98e7 cc4b72f 41a98e7 cc4b72f 41a98e7 cc4b72f 41a98e7 56ac5db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 |
import gradio as gr
import os
from dotenv import load_dotenv
from tool import YouTubeTranscriptExtractor, TranscriptSummarizer, TranscriptToDocx
# Load environment variables
load_dotenv()
youtube_tool = YouTubeTranscriptExtractor()
docx_tool = TranscriptToDocx()
#summarizer_tool = TranscriptSummarizer()
import sys
print(sys.version)
def process_youtube_video(video_url, hf_api_key, gemini_api_key, model_name, existing_docx_path=None):
# Initialize tools
summarizer_tool = TranscriptSummarizer(
hf_api_key=hf_api_key,
gemini_api_key=gemini_api_key,
model_name=model_name
)
# Get video title
from pytubefix import YouTube
try:
yt = YouTube(video_url)
video_title = yt.title
except Exception:
video_title = "YouTube Video"
# Extract transcript and detect language
transcript_result = youtube_tool.forward(video_url=video_url)
# Parse the formatted string response
# Format: "LANGUAGE:lang||transcript_text"
try:
if "LANGUAGE:" in transcript_result and "||" in transcript_result:
parts = transcript_result.split("||", 1)
language = parts[0].replace("LANGUAGE:", "").strip()
transcript = parts[1]
print(f"Detected language: {language}")
else:
# Fallback if we didn't get the expected format
transcript = transcript_result
language = "en"
print("Warning: Could not detect language, using English as default")
except Exception as e:
transcript = transcript_result if isinstance(transcript_result, str) else "Error extracting transcript"
language = "en"
print(f"Warning: Error parsing transcript data: {str(e)}, using English as default")
# Generate summary and get image URL
summary_and_blog = summarizer_tool.forward(transcript=transcript, language=language)
try:
if "\n\nImage URL: " in summary_and_blog:
summary, image_url = summary_and_blog.split("\n\nImage URL: ")
else:
summary = summary_and_blog
image_url = None
except Exception:
summary = summary_and_blog
image_url = None
# Generate or update DOCX file
# Handle the file path from Gradio
docx_file_path = None
if existing_docx_path is not None and existing_docx_path != "" and existing_docx_path != []:
# If it's a temporary file path from Gradio
if isinstance(existing_docx_path, str) and os.path.exists(existing_docx_path):
docx_file_path = existing_docx_path
# If it's a file object from Gradio
elif hasattr(existing_docx_path, 'name') and os.path.exists(existing_docx_path.name):
docx_file_path = existing_docx_path.name
# If it's a list (Gradio sometimes returns a list for file components)
elif isinstance(existing_docx_path, list) and len(existing_docx_path) > 0 and existing_docx_path[0] is not None:
if isinstance(existing_docx_path[0], str) and os.path.exists(existing_docx_path[0]):
docx_file_path = existing_docx_path[0]
elif hasattr(existing_docx_path[0], 'name') and os.path.exists(existing_docx_path[0].name):
docx_file_path = existing_docx_path[0].name
docx_path = docx_tool.forward(
transcript=transcript,
summary=summary,
video_title=video_title,
image_path=image_url,
existing_docx_path=docx_file_path
)
return transcript, summary, image_url, docx_path
with gr.Blocks() as demo:
gr.Markdown("# YouTube Transcript Summarizer and Blog Content Generator")
gr.Markdown("Enter a YouTube video URL and Hugging Face API Key to extract the transcript, summarize it, and generate blog content with an image and DOCX file. Optionally, you can provide an existing DOCX file to update.")
# Information about API keys
gr.Markdown("### API Keys and Model Configuration")
gr.Markdown("You need to provide both a Hugging Face API key (for image generation) and a Gemini API key (for summarization). You can now enter your Gemini API key directly in the UI or set it in the .env file.")
with gr.Row():
with gr.Column():
video_url = gr.Textbox(label="YouTube Video URL")
# API Keys
gr.Markdown("#### API Keys")
hf_api_key = gr.Textbox(label="Hugging Face API Key", type="password")
gemini_api_key = gr.Textbox(
label="Gemini API Key",
type="password",
value=os.getenv("GEMINI_API_KEY", "")
)
# Model Selection
model_name = gr.Dropdown(
label="Gemini Model",
choices=["gemini-2.0-flash", "gemini-1.5-pro", "gemini-1.5-flash"],
value="gemini-2.0-flash"
)
existing_docx = gr.File(label="Existing DOCX file (optional)", file_types=[".docx"])
submit_btn = gr.Button("Process Video")
with gr.Column():
transcript_output = gr.Textbox(label="Transcript")
summary_output = gr.Textbox(label="Summary and Blog Content")
image_output = gr.Image(label="Generated Image", image_mode="RGBA")
docx_output = gr.File(label="Generated DOCX File")
submit_btn.click(
fn=process_youtube_video,
inputs=[video_url, hf_api_key, gemini_api_key, model_name, existing_docx],
outputs=[transcript_output, summary_output, image_output, docx_output]
)
iface = demo
iface.launch()
|