Update app.py

app.py

@@ -81,16 +81,16 @@ def generate_image(prompt, base="写实风格", motion="", step=8, progress=gr.P
 
 # Gradio Interface
 with gr.Blocks(css="style.css") as demo:
-    # 登录界面
-    with gr.Tab("登录") as login_tab:
-        gr.HTML("<h1><center>officechat视频生成系统登录</center></h1>")
+    # 创建两个界面容器
+    with gr.Group(visible=True) as login_container:
+        gr.HTML("<h1><center>视频生成系统登录</center></h1>")
         username = gr.Textbox(label="用户名")
         password = gr.Textbox(label="密码", type="password")
         login_button = gr.Button("登录")
         login_msg = gr.Textbox(label="登录状态", interactive=False)
         
     # 主界面
-    with gr.Tab("视频生成") as video_tab:
+    with gr.Group(visible=False) as main_container:
         gr.HTML(
             "<h1><center>officechat即时视频生成系统</center></h1>" +
             "<p><center><span style='color: red;'>如果生成效果不理想，可以将步数从4调整到8。</span></center></p>" +
@@ -171,22 +171,24 @@ with gr.Blocks(css="style.css") as demo:
             cache_examples="lazy",
         )
 
+        # 生成按钮事件
+        submit.click(
+            fn=generate_image,
+            inputs=[prompt, select_base, select_motion, select_step],
+            outputs=[video]
+        )
+
     # 登录逻辑
     def handle_login(username, password):
         success, message = login(username, password)
         if success:
-            video_tab.visible = True
-            login_tab.visible = False
-            return message, gr.update(selected="视频生成")
-        return message, gr.update(selected="登录")
+            return message, gr.update(visible=False), gr.update(visible=True)
+        return message, gr.update(visible=True), gr.update(visible=False)
 
     login_button.click(
         fn=handle_login,
         inputs=[username, password],
-        outputs=[login_msg, demo.select]
+        outputs=[login_msg, login_container, main_container]
     )
 
-    # 设置初始状态
-    video_tab.visible = False
-
 demo.queue().launch()