美化UI界面：添加新布局和样式

app.py

@@ -35,108 +35,24 @@ def detect_objects(image):
     except Exception as e:
         return None, f"Error: {str(e)}"
 
-# 创建主题和样式
-theme = gr.themes.Soft(
-    primary_hue="indigo",
-    secondary_hue="blue",
-).set(
-    body_background_fill="*neutral_50",
-    block_background_fill="*neutral_100",
-    block_label_background_fill="*primary_100",
-    block_label_text_color="*primary_500",
-    button_primary_background_fill="*primary_500",
-    button_primary_background_fill_hover="*primary_600",
-    button_primary_text_color="white",
-    border_color_primary="*primary_300",
-)
-
-with gr.Blocks(theme=theme) as demo:
-    gr.Markdown(
-        """
-        # 🐉 奶龙杀手 (NailongKiller)
-        
-        这是一个基于 YOLO 的奶龙检测系统。上传图片即可自动检测图中的奶龙。
-        
-        This is a YOLO-based Nailong detection system. Upload an image to detect Nailong automatically.
-        """
-    )
+# 创建Gradio界面
+demo = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="numpy", label="输入图片 | Input Image"),
+    outputs=[
+        gr.Image(type="numpy", label="检测结果 | Detection Result"),
+        gr.Textbox(label="检测信息 | Detection Info")
+    ],
+    title="🐉 奶龙杀手 (NailongKiller)",
+    description="""
+    这是一个基于 YOLO 的奶龙检测系统。上传图片即可自动检测图中的奶龙。
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            input_image = gr.Image(
-                label="输入图片 | Input Image",
-                type="numpy",
-                height=512,
-                width=512,
-            )
-            
-            with gr.Row():
-                clear_btn = gr.Button("清除 | Clear", variant="secondary", size="lg")
-                detect_btn = gr.Button("检测 | Detect", variant="primary", size="lg")
-        
-        with gr.Column(scale=1):
-            output_image = gr.Image(
-                label="检测结果 | Detection Result",
-                height=512,
-                width=512,
-            )
-            result_text = gr.Textbox(
-                label="检测信息 | Detection Info",
-                placeholder="等待检测...",
-            )
-    
-    gr.Markdown(
-        """
-        ### 📝 使用说明 | Instructions
-        
-        1. 点击上传或拖拽图片到左侧输入区域
-        2. 点击"检测"按钮开始识别
-        3. 右侧将显示检测结果和统计信息
-        
-        ### ⚠️ 注意事项 | Notes
-        
-        - 支持常见图片格式 (jpg, png, etc.)
-        - 建议上传清晰的图片以获得更好的检测效果
-        - 图片会自动调整大小以优化性能
-        
-        ### 🔗 相关链接 | Links
-        
-        - [项目地址 | Project Repository](https://huggingface.co/spaces/Hakureirm/NailongKiller)
-        - [YOLO Documentation](https://docs.ultralytics.com/)
-        """
-    )
-    
-    # 事件处理
-    detect_btn.click(
-        fn=detect_objects,
-        inputs=input_image,
-        outputs=[output_image, result_text],
-    )
-    
-    clear_btn.click(
-        lambda: (None, None, None),
-        outputs=[input_image, output_image, result_text],
-    )
-    
-    # 修改示例部分的代码
-    # 使用try-except来处理示例图片
-    try:
-        examples = []
-        if os.path.exists("example1.jpg"):
-            examples.append("example1.jpg")
-        if os.path.exists("example2.jpg"):
-            examples.append("example2.jpg")
-            
-        if examples:  # 只有在有示例图片时才添加Examples组件
-            gr.Examples(
-                examples=examples,
-                inputs=input_image,
-                outputs=[output_image, result_text],
-                fn=detect_objects,
-                cache_examples=True,
-            )
-    except Exception as e:
-        print(f"Warning: Could not load example images: {e}")
+    This is a YOLO-based Nailong detection system. Upload an image to detect Nailong automatically.
+    """,
+    theme=gr.themes.Default(),
+    allow_flagging="never",
+    examples=["example1.jpg", "example2.jpg"] if all(os.path.exists(f) for f in ["example1.jpg", "example2.jpg"]) else None
+)
 
 # API端点
 @app.post("/detect/")