Update app.py

app.py

@@ -3,83 +3,88 @@ import os
 import yaml
 import shutil
 import sys
+from datetime import datetime
 
-# 强制在启动时打印 START，并直接输出到 stderr 确保可见
-print("==== [SYSTEM] APP STARTING... ====", file=sys.stderr, flush=True)
+# 强制刷新打印（最后一次尝试后台日志）
+print("!!! CRITICAL STARTUP CHECK !!!", file=sys.stderr, flush=True)
 
-# 确保 papers 文件夹存在
-os.makedirs("papers", exist_ok=True)
+# 初始化环境
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+PAPERS_DIR = os.path.join(BASE_DIR, "papers")
+CONFIG_PATH = os.path.join(BASE_DIR, "config.yaml")
+os.makedirs(PAPERS_DIR, exist_ok=True)
 
-def debug_log(message):
-    """自定义调试函数，确保内容立即出现在日志中"""
-    print(f"[DEBUG] {message}", file=sys.stderr, flush=True)
+def get_debug_info():
+    """获取当前服务器文件状态的快照"""
+    now = datetime.now().strftime("%H:%M:%S")
+    files = os.listdir(PAPERS_DIR) if os.path.exists(PAPERS_DIR) else "Directory missing"
+    
+    config_content = "Not found"
+    if os.path.exists(CONFIG_PATH):
+        with open(CONFIG_PATH, "r", encoding="utf-8") as f:
+            config_content = f.read()
+            
+    return f"[{now}] 📁 Files in papers/: {files}\n\n[{now}] 📄 config.yaml content:\n{config_content}"
 
 def save_pdf(file):
     if file is None:
-        return "Please upload a PDF file."
-    
-    debug_log(f"Received file: {file.name}")
+        return "Please upload a PDF.", get_debug_info()
     
-    # 获取文件名并保存
-    file_name = os.path.basename(file.name)
-    file_path = os.path.join("papers", file_name)
+    file_path = os.path.join(PAPERS_DIR, os.path.basename(file.name))
     shutil.copy(file.name, file_path)
-    
-    # 打印文件夹下的所有文件
-    all_files = os.listdir("papers")
-    debug_log(f"File saved to {file_path}")
-    debug_log(f"Current files in 'papers/': {all_files}")
-    
-    return f"File '{file_name}' saved. Total files in folder: {len(all_files)}"
+    return f"✅ Saved {os.path.basename(file.name)}", get_debug_info()
 
 def save_api_key(api_key):
     if not api_key:
-        return "API Key cannot be empty."
-    
-    config_path = "config.yaml"
-    debug_log("Attempting to save API Key...")
+        return "API Key is empty.", get_debug_info()
     
     try:
         config = {}
-        if os.path.exists(config_path):
-            with open(config_path, "r", encoding="utf-8") as f:
+        if os.path.exists(CONFIG_PATH):
+            with open(CONFIG_PATH, "r", encoding="utf-8") as f:
                 config = yaml.safe_load(f) or {}
         
-        if "api_keys" not in config:
-            config["api_keys"] = {}
-        
+        if "api_keys" not in config: config["api_keys"] = {}
         config["api_keys"]["gemini_api_key"] = api_key
         
-        with open(config_path, "w", encoding="utf-8") as f:
-            yaml.dump(config, f, allow_unicode=True, default_flow_style=False)
-        
-        # 验证并打印 config 内容 (脱敏处理)
-        with open(config_path, "r", encoding="utf-8") as f:
-            current_config = f.read()
-            # 打印前20个字符确认文件结构，不打印完整 key
-            debug_log(f"Config updated. Content preview: {current_config[:20]}...")
+        with open(CONFIG_PATH, "w", encoding="utf-8") as f:
+            yaml.dump(config, f, allow_unicode=True)
             
-        return "Gemini API Key updated successfully!"
+        return "✅ API Key saved!", get_debug_info()
     except Exception as e:
-        debug_log(f"ERROR: {str(e)}")
-        return f"Error: {str(e)}"
+        return f"❌ Error: {str(e)}", get_debug_info()
 
-with gr.Blocks(title="PDF Uploader & Config") as demo:
-    gr.Markdown("# PDF and API Configuration")
+# --- UI 界面 ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🛠️ Space Files Debugger")
     
-    with gr.Tab("Upload PDF"):
-        file_input = gr.File(label="Upload your PDF", file_types=[".pdf"])
-        upload_btn = gr.Button("Save PDF")
-        upload_output = gr.Textbox(label="Status")
-        upload_btn.click(save_pdf, inputs=file_input, outputs=upload_output)
+    with gr.Row():
+        with gr.Column(scale=2):
+            with gr.Tab("1. Upload PDF"):
+                pdf_input = gr.File(label="Upload PDF", file_types=[".pdf"])
+                pdf_btn = gr.Button("Save PDF")
+                pdf_msg = gr.Textbox(label="Status")
+                
+            with gr.Tab("2. Config"):
+                key_input = gr.Textbox(label="Gemini API Key", type="password")
+                key_btn = gr.Button("Save Key")
+                key_msg = gr.Textbox(label="Status")
         
-    with gr.Tab("Gemini Configuration"):
-        key_input = gr.Textbox(label="Enter Gemini API Key", type="password")
-        key_btn = gr.Button("Save API Key")
-        key_output = gr.Textbox(label="Status")
-        key_btn.click(save_api_key, inputs=key_input, outputs=key_output)
+        # 右侧调试面板��实时显示服务器文件系统状态
+        with gr.Column(scale=1):
+            gr.Markdown("### 🔍 System Real-time Monitor")
+            debug_view = gr.Textbox(
+                label="Server File Explorer", 
+                value=get_debug_info(), 
+                lines=15,
+                interactive=False
+            )
+            refresh_btn = gr.Button("🔄 Force Refresh View")
+
+    # 绑定逻辑：每次操作都更新右侧的调试面板
+    pdf_btn.click(save_pdf, inputs=pdf_input, outputs=[pdf_msg, debug_view])
+    key_btn.click(save_api_key, inputs=key_input, outputs=[key_msg, debug_view])
+    refresh_btn.click(get_debug_info, outputs=debug_view)
 
 if __name__ == "__main__":
-    debug_log("Gradio Interface launching...")
-    # HF Spaces 环境下 server_name 和 port 通常由平台处理，简单 launch 即可
     demo.launch()