verify_password(password_secret, correct_password)

app.py

@@ -6,6 +6,16 @@ import yt_dlp as ydlp
 from openai import OpenAI
 import re
 
+
+OPEN_AI_KEY = os.environ.get("OPEN_AI_KEY")  # 從環境變量中獲取默認的OpenAI key
+PASSWORD = os.environ.get("PASSWORD_SECRET")
+
+def verify_password(input_password, correct_password):
+    if input_password == correct_password:
+        return True
+    else:
+        raise gr.Error("密碼錯誤")
+
 def ms_to_srt_time(ms):
     sec, ms = divmod(ms, 1000)
     min, sec = divmod(sec, 60)
@@ -30,20 +40,15 @@ def get_video_duration(url):
         info_dict = ydl.extract_info(url, download=False)
         return info_dict.get('duration', 0)
 
-def process_video(yt_id_or_url, openAI_key=None, password_secret=None):
+def process_video(yt_id_or_url, openAI_key=None, password_secret=None):    
     # Extract the yt_id from the URL if a full URL is given
     yt_id_match = re.search(r"(?<=v=)[a-zA-Z0-9_-]+", yt_id_or_url)
     yt_id = yt_id_match.group(0) if yt_id_match else yt_id_or_url
 
     # Check if openAI_key is provided or validate using secret
     if not openAI_key:
-        default_key = os.environ.get("OPEN_AI_KEY")  # 從環境變量中獲取默認的OpenAI key
-        default_secret = os.environ.get("PASSWORD_SECRET")
-        
-        if password_secret == default_secret:
-            openAI_key = default_key
-        else:
-            return "Invalid secret or no OpenAI key provided."
+        correct_password = PASSWORD
+        verify_password(password_secret, correct_password)
 
     # yt_id = "90BAlvlLvE0"
     url = f"https://www.youtube.com/watch?v={yt_id}"
@@ -253,26 +258,26 @@ def process_video(yt_id_or_url, openAI_key=None, password_secret=None):
 
 
 
-def main():
-    interface = gr.Interface(
+with gr.Blocks() as demo:
+    with gr.Row():
+        video_id = gr.Textbox(label="YouTube Video ID")
+        openai_key = gr.Textbox(label="OpenAI Key (optional)")
+        password_secret = gr.Textbox(label="Password Secret (optional)")
+    with gr.Row():
+        download_srt = gr.File(label="Download SRT")
+        srt_content = gr.Textbox(label="SRT Content", show_copy_button=True)
+        large_scope_srt = gr.Textbox(label="Large Scope SRT", show_copy_button=True)
+        video_summary = gr.Textbox(label="Video Summary", show_copy_button=True)
+        mind_map = gr.Textbox(label="Mind Map", show_copy_button=True)
+    
+    gr.Markdown("Generate SRT, Summary and Mind Map from YouTube video（限額 60 min）")
+    gr.Markdown("YouTube to SRT, Summary & Mind Map")
+    submit_btn = gr.Button("Process Video")
+    submit_btn.click(
         fn=process_video,
-        inputs=[
-            gr.Textbox(label="YouTube Video ID"),
-            gr.Textbox(label="OpenAI Key (optional)"),
-            gr.Textbox(label="Password Secret (optional)")
-        ],
-        outputs=[
-            gr.File(label="Download SRT"),
-            gr.Textbox(label="SRT Content",  show_copy_button=True),
-            gr.Textbox(label="Large Scope SRT", show_copy_button=True),
-            gr.Textbox(label="Video Summary", show_copy_button=True),
-            gr.Textbox(label="Mind Map", show_copy_button=True)
-        ],
-        live=False,
-        description="Generate SRT, Summary and Mind Map from YouTube video（限額 60 min）",
-        title="YouTube to SRT, Summary & Mind Map"
+        inputs=[video_id, openai_key, password_secret],
+        outputs=[download_srt, srt_content, large_scope_srt, video_summary, mind_map]
     )
-    interface.launch()
 
-if __name__ == "__main__":
-    main()
+
+demo.launch()