Update app.py

app.py

@@ -19,7 +19,6 @@ def load_model():
         try:
             from transformers import SamModel, SamProcessor
             
-            # UPGRADE: Large Model für bessere Qualität
             model_name = "facebook/sam-vit-large"
             
             processor = SamProcessor.from_pretrained(model_name)
@@ -55,18 +54,13 @@ def prepare_image(image, max_size=1024):
     return image_pil, image_np
 
 def refine_mask(mask, kernel_size=5):
-    """Maske glätten und Löcher füllen"""
     mask_uint8 = (mask > 0).astype(np.uint8) * 255
-    
     kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (kernel_size, kernel_size))
     mask_closed = cv2.morphologyEx(mask_uint8, cv2.MORPH_CLOSE, kernel)
     mask_refined = cv2.morphologyEx(mask_closed, cv2.MORPH_OPEN, kernel)
-    
     return mask_refined > 0
 
 def segment_automatic(image, quality="high"):
-    """Automatische Segmentierung mit Multi-Mask Output"""
-    
     if image is None:
         return None, {"error": "Kein Bild hochgeladen"}
     
@@ -157,8 +151,6 @@ def segment_automatic(image, quality="high"):
         return image, {"error": str(e)}
 
 def segment_multi_dense(image, density="medium"):
-    """Multi-Object mit dichterem Grid"""
-    
     if image is None:
         return None, {"error": "Kein Bild"}
     
@@ -259,12 +251,13 @@ def segment_multi_dense(image, density="medium"):
         return image, {"error": str(e)}
 
 # Gradio Interface
-with gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft()) as demo:
-    
+demo = gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft())
+
+with demo:
     gr.Markdown("# 🎨 SAM2 Segmentierung v2.0")
-    gr.Markdown("### Verbesserte Qualität mit Large Model & Multi-Mask Output")
+    gr.Markdown("### Verbesserte Qualität mit Large Model")
     
-    with gr.Tab("🤖 Automatisch (Verbessert)"):
+    with gr.Tab("🤖 Automatisch"):
         with gr.Row():
             with gr.Column():
                 input_auto = gr.Image(type="pil", label="Bild hochladen")
@@ -274,8 +267,6 @@ with gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft()) as demo:
                     label="Qualität"
                 )
                 btn_auto = gr.Button("Segmentieren", variant="primary", size="lg")
-                
-                gr.Markdown("**Verbesserungen:** Large Model, Multi-Mask, Kanten-Glättung")
             
             with gr.Column():
                 output_auto = gr.Image(label="Ergebnis")
@@ -287,7 +278,7 @@ with gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft()) as demo:
             outputs=[output_auto, json_auto]
         )
     
-    with gr.Tab("🎯 Multi-Region (Dicht)"):
+    with gr.Tab("🎯 Multi-Region"):
         with gr.Row():
             with gr.Column():
                 input_multi = gr.Image(type="pil", label="Bild hochladen")
@@ -297,8 +288,6 @@ with gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft()) as demo:
                     label="Punkt-Dichte"
                 )
                 btn_multi = gr.Button("Alle Bereiche", variant="primary", size="lg")
-                
-                gr.Markdown("**Grid:** High=5x5, Medium=4x4, Low=3x3")
             
             with gr.Column():
                 output_multi = gr.Image(label="Ergebnis")
@@ -310,10 +299,28 @@ with gr.Blocks(title="SAM2 v2.0", theme=gr.themes.Soft()) as demo:
             outputs=[output_multi, json_multi]
         )
     
-    with gr.Tab("📡 API Info"):
-        api_doc = """
-### API Endpoint
-
-**URL:** https://dein-space.hf.space/api/predict
+    with gr.Tab("📡 API"):
+        gr.Markdown("### API Endpoint")
+        gr.Markdown("URL: https://EnginDev-Boostly.hf.space/api/predict")
+        gr.Markdown("### JavaScript Beispiel")
+        gr.Code('''
+fetch('https://EnginDev-Boostly.hf.space/api/predict', {
+    method: 'POST',
+    headers: {'Content-Type': 'application/json'},
+    body: JSON.stringify({
+        data: [imageBase64, "high"],
+        fn_index: 0
+    })
+})
+.then(r => r.json())
+.then(result => console.log(result.data));
+        ''', language="javascript")
+        
+        gr.Markdown("### Parameter")
+        gr.Markdown("- fn_index: 0 = Automatisch, 1 = Multi-Region")
+        gr.Markdown("- quality: high oder fast")
+        gr.Markdown("- density: high, medium, low")
 
-**JavaScript Beispiel:**
+if __name__ == "__main__":
+    print("🌐 Launching v2.0...")
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)