Create app.py

app.py

@@ -0,0 +1,79 @@
+import subprocess
+import sys
+import os
+import random
+import gradio as gr
+from pydub import AudioSegment
+import google.generativeai as genai
+from google.colab import userdata
+from youtube_transcript_api import YouTubeTranscriptApi
+
+try:
+    import openai
+except ImportError:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "openai"])
+    import openai  # Import the library after installing it
+
+def DownloadScript(youtube_url, key):
+      video_id = youtube_url.split("v=")[-1]
+      transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['en'])
+      full_text = " ".join([entry['text'] for entry in transcript])
+      return full_text
+
+def GenQuiz(text, key):
+      prompt = "請依附內容，為我出五題選擇題"+text
+      openai.api_key = key
+      completion = openai.chat.completions.create(
+          model="gpt-4o",
+          messages=[{"role": "user", "content": prompt}]
+          )
+      return completion.choices[0].message.content
+
+def setup_gradio_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown(
+            """
+            <h1 style="text-align: center; font-size: 36px; color: #333;">YouTube 字幕考題生成器</h1>
+            """
+        )
+        with gr.Tab("字幕考題生成"):
+            api_key_input = gr.Textbox(label="第一步：請輸入OpenAI API金鑰", placeholder="OpenAI API Key")
+            youtube_input = gr.Textbox(
+                label="第二步：請輸入 YouTube 影片連結", 
+                placeholder="https://www.youtube.com/watch?v=example"
+            )
+            submit_button = gr.Button("載入字幕")
+            
+            txt_cc_output = gr.Textbox(
+                label="第三步：下載字幕", 
+                placeholder="字幕將顯示在此處", 
+                interactive=False
+            )
+
+            submit2_button = gr.Button("考題生成")
+            
+            txt_quiz_output = gr.Textbox(
+                label="第四步：考題生成", 
+                placeholder="考題將顯示在此處", 
+                interactive=False
+            )
+
+        # 連接按鈕到功能
+        submit_button.click(
+            DownloadScript,
+            inputs=[youtube_input, api_key_input],
+            outputs=[txt_cc_output]
+        )
+
+        submit2_button.click(
+            GenQuiz,
+            inputs=[txt_cc_output, api_key_input],
+            outputs=[txt_quiz_output]
+        )
+
+    return demo
+
+# 執行界面
+if __name__ == "__main__":
+    demo = setup_gradio_interface()
+    demo.launch()