Update app.py

app.py

@@ -952,6 +952,62 @@ def handle_user_access_final(request: gr.Request):
         """
         return title, gr.HTML(error_html)
 
+# 添加可选的管理界面（用于调试）
+def create_admin_interface():
+    """创建管理界面（可选）- 简化版本"""
+    with gr.Blocks(title="CIV3283 Load Distributor - Admin") as admin_interface:
+        gr.Markdown("# 🔧 CIV3283 Load Distributor - Admin Panel")
+        
+        with gr.Row():
+            cache_status_btn = gr.Button("📊 Check Cache Status", variant="secondary")
+            force_update_btn = gr.Button("🔄 Force Cache Update", variant="primary")
+        
+        cache_info_display = gr.JSON(label="Cache Information")
+        status_message = gr.Markdown("")
+        
+        def check_cache_status():
+            try:
+                status = get_cache_status()
+                if 'error' in status:
+                    return status, "❌ Cache not available"
+                else:
+                    age_min = status.get('age_minutes', 0)
+                    spaces_count = status.get('spaces_count', 0)
+                    is_fresh = status.get('is_fresh', False)
+                    
+                    if is_fresh:
+                        message = f"✅ Cache is fresh ({age_min:.1f} min old, {spaces_count} spaces)"
+                    else:
+                        message = f"⚠️ Cache is stale ({age_min:.1f} min old, {spaces_count} spaces)"
+                    
+                    return status, message
+            except Exception as e:
+                return {"error": str(e)}, f"❌ Error checking cache: {str(e)}"
+        
+        def force_update():
+            try:
+                result = force_cache_update()
+                if 'error' in result:
+                    return result, "❌ Failed to update cache"
+                else:
+                    # 获取更新后的状态
+                    status = get_cache_status()
+                    return status, "✅ Cache updated successfully"
+            except Exception as e:
+                return {"error": str(e)}, f"❌ Error updating cache: {str(e)}"
+        
+        cache_status_btn.click(
+            fn=check_cache_status,
+            outputs=[cache_info_display, status_message]
+        )
+        
+        force_update_btn.click(
+            fn=force_update,
+            outputs=[cache_info_display, status_message]
+        )
+    
+    return admin_interface
+
 # 修改后的主Gradio界面
 def create_main_interface():
     """创建主要的用户界面"""
@@ -959,21 +1015,7 @@ def create_main_interface():
         title_display = gr.Markdown("# 🔄 CIV3283 Learning Assistant Load Distributor", elem_id="title")
         content_display = gr.HTML("")
         
-        # 添加隐藏的缓存信息显示（用于调试，默认不显示）
-        with gr.Accordion("🔧 System Information", open=False, visible=False):
-            cache_info = gr.JSON(label="Cache Status", visible=False)
-            
-            def update_cache_info():
-                return get_cache_status()
-            
-            # 每30秒更新一次缓存信息（如果界面是打开的）
-            interface.load(
-                fn=update_cache_info,
-                outputs=cache_info,
-                every=30
-            )
-        
-        # 主要的负载均衡逻辑
+        # 主要的负载均衡逻辑 - 页面加载时执行
         interface.load(
             fn=handle_user_access_final,
             outputs=[title_display, content_display]
@@ -983,21 +1025,26 @@ def create_main_interface():
 
 # 修改后的main函数
 if __name__ == "__main__":
-    # 创建主界面
-    main_interface = create_main_interface()
-      
-    # 启动主界面
-    main_interface.launch()
-
-# 在程序退出时优雅地关闭后台线程
-import atexit
-
-def cleanup_on_exit():
-    """程序退出时的清理函数"""
-    global activity_cache
-    if activity_cache:
-        print("[cleanup_on_exit] Stopping background cache updates...")
-        activity_cache.stop_background_updates()
-
-# 注册退出时的清理函数
-atexit.register(cleanup_on_exit)
+    import sys
+    
+    # 检查是否需要管理界面
+    enable_admin = "--admin" in sys.argv or os.environ.get("ENABLE_ADMIN", "false").lower() == "true"
+    
+    if enable_admin:
+        print("🔧 Starting with admin interface enabled")
+        # 创建带管理界面的组合界面
+        main_interface = create_main_interface()
+        admin_interface = create_admin_interface()
+        
+        # 使用TabbedInterface组合两个界面
+        combined_interface = gr.TabbedInterface(
+            [main_interface, admin_interface],
+            ["🎯 Load Distributor", "🔧 Admin Panel"],
+            title="CIV3283 Load Distributor System"
+        )
+        combined_interface.launch()
+    else:
+        print("🎯 Starting main interface only")
+        # 只启动主界面
+        main_interface = create_main_interface()
+        main_interface.launch()