password

app.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 
 from typing import Container
+from config.config import PASSWORD
 import gradio as gr
 import os
 import shutil
@@ -8,12 +9,14 @@ import tempfile
 from google import genai
 from google.genai import types
 
-from initializer import initialize_clients
+from initializer import initialize_clients, initialize_password
 
 # 初始化 Google Cloud Storage 服務和 GENAI 客戶端
 GCS_SERVICE, GENAI_CLIENT = initialize_clients()
 GCS_CLIENT = GCS_SERVICE.client
 
+PASSWORD = initialize_password()
+
 def toggle_visibility(toggle_value):
     return gr.update(visible=toggle_value)
 
@@ -175,6 +178,14 @@ def process_all_files(file_list):
     print("\n=== 檔案處理完成 ===")
     return result_text, transcript_text
 
+def process_with_auth(password, file_list):
+    """包含密碼驗證的處理函數"""
+    if not password or password != PASSWORD:
+        return "請輸入正確的密碼", "", gr.update(visible=False)
+    
+    result_text, transcript_text = process_all_files(file_list)
+    return result_text, transcript_text
+
 def on_summary_click(transcript):
     if not transcript:
         return "請先上傳文件或輸入 YouTube 連結並處理完成後再生成摘要。"
@@ -183,7 +194,10 @@ def on_summary_click(transcript):
     return summary
 
 with gr.Blocks() as demo:
-    gr.Markdown("# AI Notes Assistant")
+
+    with gr.Row():
+        gr.Markdown("# AI Notes Assistant")
+        password_input = gr.Textbox(label="password")
 
     with gr.Row():
         source_toggle = gr.Checkbox(label="顯示來源選單", value=True)
@@ -209,8 +223,8 @@ with gr.Blocks() as demo:
             file_display = gr.CheckboxGroup(label="已上傳的文件", interactive=True)
 
             process_files_button = gr.Button("處理檔案")
-            rag_result = gr.Textbox(label="處理結果", interactive=False)
-
+            rag_result = gr.Textbox(label="處理狀態", interactive=False)
+            
             file_list.change(lambda x: gr.update(choices = [os.path.basename(path) if os.path.basename(path) else path for path in x]), inputs=file_list, outputs=file_display)
 
         with gr.Column(visible=True) as chat_column:
@@ -244,13 +258,14 @@ with gr.Blocks() as demo:
     chat_toggle.change(toggle_visibility, inputs=chat_toggle, outputs=chat_column)
     feature_toggle.change(toggle_visibility, inputs=feature_toggle, outputs=feature_column)
 
-
-
     # 更新處理檔案按鈕的事件處理
     process_files_button.click(
-        process_all_files,
-        inputs=[file_list],
-        outputs=[rag_result, transcript_display]
+        fn=process_with_auth,
+        inputs=[password_input, file_list],
+        outputs=[
+            rag_result, 
+            transcript_display
+        ]
     ).then(
         fn=on_summary_click,
         inputs=[transcript_display],

initializer.py

@@ -67,9 +67,31 @@ def initialize_genai_client(google_credentials_key):
             location='us-central1'
         )
 
+def initialize_password():
+    is_env_local = os.getenv("IS_ENV_LOCAL", "false") == "true"
+    print(f"Environment is local: {is_env_local}")
+    
+    try:
+        if is_env_local:
+            config_path = os.path.join(os.path.dirname(__file__), "local_config.json")
+            print(f"Trying to load config from: {config_path}")
+            if not os.path.exists(config_path):
+                print(f"Warning: {config_path} does not exist")
+                return None
+            
+            with open(config_path) as f:
+                config = json.load(f)
+                return config["PASSWORD"]
+        else:
+            return os.getenv("PASSWORD")
+    except Exception as e:
+        print(f"Error initializing password: {str(e)}")
+        return None
+
 def initialize_clients():
     google_credentials_key = initialize_google_credentials()
     gcs_service = initialize_gcs_service(google_credentials_key)
     genai_client = initialize_genai_client(google_credentials_key)
     
-    return gcs_service, genai_client
+    return gcs_service, genai_client
+