Upload 2 files

app.py

@@ -0,0 +1,107 @@
+# app.py
+import os
+import subprocess
+import glob
+import re
+import traceback
+import gradio as gr
+from openai import OpenAI
+
+# Load OpenAI key from environment (Hugging Face Spaces -> Settings -> Secrets)
+openai_api_key = os.getenv("OPENAI_API_KEY")
+openai = OpenAI(api_key=openai_api_key)
+
+def download_audio(youtube_url):
+    try:
+        output_template = "/tmp/downloaded_audio.%(ext)s"
+
+        # Remove any old files
+        for f in glob.glob("/tmp/downloaded_audio.*"):
+            os.remove(f)
+
+        command = [
+            "yt-dlp", "-f", "bestaudio",
+            "--extract-audio", "--audio-format", "mp3",
+            "--audio-quality", "0",
+            "-o", output_template,
+            youtube_url
+        ]
+
+        result = subprocess.run(command, capture_output=True, text=True)
+        print("stdout:\n", result.stdout)
+        print("stderr:\n", result.stderr)
+
+        if result.returncode != 0:
+            raise RuntimeError(f"yt-dlp failed: {result.stderr}")
+
+        files = glob.glob("/tmp/downloaded_audio.*")
+        if not files:
+            raise FileNotFoundError("No audio file downloaded.")
+
+        return files[0]
+    except Exception as e:
+        raise RuntimeError(f"Download error: {e}")
+
+def transcribe_audio(file_path):
+    try:
+        with open(file_path, "rb") as f:
+            result = openai.audio.transcriptions.create(
+                model="whisper-1",
+                file=f,
+                response_format="verbose_json"
+            )
+        return result["text"], result["language"]
+    except Exception as e:
+        raise RuntimeError(f"Transcription error: {e}")
+
+def summarize_text(text, lang):
+    lang = lang.lower()
+    if lang.startswith("zh") or "chinese" in lang:
+        prompt = "你是一位聰明的助手，能夠用繁體中文清楚且完整地摘要影片內容。"
+    elif lang.startswith("ja") or "japanese" in lang:
+        prompt = "あなたは日本語で要点を簡潔かつ分かりやすく要約する有能なアシスタントです。"
+    else:
+        prompt = "You are a helpful assistant that summarizes transcripts clearly and concisely."
+
+    response = openai.chat.completions.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": prompt},
+            {"role": "user", "content": f"Summarize the following transcript:\n\n{text}"}
+        ]
+    )
+
+    summary = response.choices[0].message.content
+    debug_info = f"🌐 Detected Language: {lang}\n🧠 Prompt Used: {prompt}"
+    return summary, debug_info
+
+def extract_video_id(url):
+    match = re.search(r"(?:v=|shorts/)([a-zA-Z0-9_-]{11})", url)
+    return match.group(1) if match else None
+
+def full_process(youtube_url):
+    try:
+        video_id = extract_video_id(youtube_url)
+        thumbnail_url = f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg" if video_id else None
+        audio_path = download_audio(youtube_url)
+        transcript, lang = transcribe_audio(audio_path)
+        summary, debug = summarize_text(transcript, lang)
+        return summary, debug, thumbnail_url
+    except Exception as e:
+        return f"❌ Error: {str(e)}", "", None
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 YouTube AI Summarizer\nEasily extract summaries from YouTube videos using Whisper + GPT. Supports English/Japanese/Chinese.")
+
+    with gr.Row():
+        youtube_input = gr.Textbox(label="🎥 Enter YouTube Video Link")
+        submit_btn = gr.Button("🔍 Summarize")
+
+    summary_output = gr.Textbox(label="📝 AI Video Summary", lines=6)
+    info_output = gr.Textbox(label="📄 Language & Model Info", lines=4)
+    thumbnail_output = gr.Image(label="🎞️ Video Thumbnail", visible=True)
+
+    submit_btn.click(fn=full_process, inputs=youtube_input, outputs=[summary_output, info_output, thumbnail_output])
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+openai
+yt-dlp
+ffmpeg-python