Update app.py

app.py

@@ -415,4 +415,43 @@ if __name__ == "__main__":
         server_port=7860, 
         share=True,
         favicon_path="favicon.ico" if os.path.exists("favicon.ico") else None
-    )
+    )
+    # 在 create_gradio_interface() 函数中添加内存管理按钮
+def create_gradio_interface():
+    with gr.Blocks(title="AI时尚设计师", theme="soft") as demo:
+        # ... [现有代码] ...
+        
+        # 添加内存管理部分
+        with gr.Row():
+            gr.Markdown("### 内存管理")
+            unload_caption_btn = gr.Button("卸载描述模型")
+            unload_sd_btn = gr.Button("卸载设计模型")
+            unload_controlnet_btn = gr.Button("卸载3D模型")
+            full_cleanup_btn = gr.Button("清理所有模型", variant="stop")
+        
+        # 内存管理事件
+        unload_caption_btn.click(
+            fn=lambda: model_manager.unload_model("caption"),
+            inputs=[],
+            outputs=[]
+        )
+        
+        unload_sd_btn.click(
+            fn=lambda: model_manager.unload_model("sd"),
+            inputs=[],
+            outputs=[]
+        )
+        
+        unload_controlnet_btn.click(
+            fn=lambda: model_manager.unload_model("controlnet"),
+            inputs=[],
+            outputs=[]
+        )
+        
+        full_cleanup_btn.click(
+            fn=model_manager.cleanup,
+            inputs=[],
+            outputs=[]
+        )
+        
+    return demo