Update app.py

app.py

@@ -1,143 +1,55 @@
 import gradio as gr
-from gradio_client import Client, handle_file
-
-# ────────────────────────────────────────────────
-#  Konfiguration der drei Spaces
-# ────────────────────────────────────────────────
-spaces = {
-    "Omni Image Editor": {
-        "url": "https://selfit-camera-omni-image-editor.hf.space/",
-        "api_name": "/text_to_image_interface",
-        "inputs": ["prompt", "aspect_ratio"],
-        "default": {"aspect_ratio": "16:9"},
-        "needs_image": False,
-        "note": "Text-to-Image – braucht nur Prompt"
-    },
-    "FireRed Image Edit": {
-        "url": "https://prithivmlmods-firered-image-edit-1-0-fast.hf.space/",
-        "api_name": "/infer",
-        "inputs": ["images", "prompt", "steps"], # Changed 'image' to 'images' (plural)
-        "default": {"steps": 28},
-        "needs_image": True,
-        "note": "Image-to-Image Edit – braucht Referenzbild + Prompt"
-    },
-    "Ana2": {
-        "url": "https://galaxydude2-ana2.hf.space/",
-        "api_name": "/predict",
-        "inputs": ["image"],
-        "default": {},
-        "needs_image": True,
-        "note": "Analyzer (Nudity/NSFW?) – braucht nur Bild, kein Prompt"
-    }
-}
-
-# ────────────────────────────────────────────────
-#  Hauptfunktion – je nach Space unterschiedlicher Aufruf
-# ────────────────────────────────────────────────
-def run_space(selected_space, prompt_text, image_input=None):
-    if selected_space not in spaces:
-        return None, "Ungültiger Space ausgewählt"
-
-    cfg = spaces[selected_space]
-    status = f"→ {selected_space} | "
-
-    # Prüfen ob Bild benötigt wird
-    if cfg["needs_image"] and image_input is None:
-        return None, status + "Dieser Space braucht ein Referenzbild"
+from gradio_client import Client
 
+def generate_omni_image(prompt, aspect_ratio):
+    client = Client("selfit-camera/Omni-Image-Editor")
     try:
-        client = Client(cfg["url"])
-
-        kwargs = cfg["default"].copy()
-        args_for_predict = [] # This list will remain empty with the current logic
-
-        # Prompt nur bei Spaces übergeben, die ihn erwarten
-        if "prompt" in cfg["inputs"] and prompt_text.strip():
-            kwargs["prompt"] = prompt_text.strip()
-
-        # Bild übergeben, wenn nötig
-        if cfg["needs_image"] and image_input:
-            processed_image = handle_file(image_input)
-            if selected_space == "Ana2":
-                # For Ana2, pass the image as a keyword argument named 'input_image_paths'
-                kwargs["input_image_paths"] = processed_image
-            elif selected_space == "FireRed Image Edit":
-                # For FireRed, pass a list of dictionaries with 'image' key
-                kwargs["images"] = [{'image': processed_image}]
+        # Ensure aspect_ratio is a string for the API
+        result = client.predict(
+           prompt=prompt,
+           aspect_ratio=aspect_ratio,
+           api_name="/text_to_image_interface"
+        )
+        # The API returns a tuple, with the image URL usually as the first element
+        if isinstance(result, (list, tuple)) and result:
+            # The first element is often the image in HTML format or a direct URL
+            # Let's try to extract a direct image URL if possible from the HTML string
+            image_output = result[0]
+            if isinstance(image_output, str) and "<img src='" in image_output:
+                start = image_output.find("src='") + len("src='")
+                end = image_output.find("'", start)
+                image_url = image_output[start:end]
+                return image_url
             else:
-                # For other spaces, pass the image as a keyword argument named 'image'
-                kwargs["image"] = processed_image
-
-        # Aspect Ratio nur bei Omni
-        if selected_space == "Omni Image Editor" and "aspect_ratio" in cfg["inputs"]:
-            kwargs["aspect_ratio"] = "16:9"   # fest codiert wie gewünscht
-
-        # API-Aufruf
-        # args_for_predict is always empty in this implementation, so we can remove it.
-        result = client.predict(**kwargs, api_name=cfg["api_name"])
-
-        # Ergebnis verarbeiten
-        if isinstance(result, (list, tuple)):
-            # häufigstes Muster: (image, status_str, ...) oder nur image
-            for item in result:
-                if hasattr(item, 'save'):  # PIL Image
-                    return item, status + "Erfolg"
-                if isinstance(item, str) and len(item) > 10 and "http" in item:
-                    return item, status + "Erfolg (Link)"
-        elif hasattr(result, 'save'):  # direkt PIL
-            return result, status + "Erfolg"
-        else:
-            return None, status + f"Unbekanntes Ergebnis-Format: {type(result)}"
-
+                return image_output # Return directly if it's a file path or URL
+        return None
     except Exception as e:
-        err_msg = str(e).replace(cfg["url"], "[URL]").replace(cfg["api_name"], "[API]")
-        return None, status + f"Fehler: {err_msg}"
+        return f"Error generating image: {e}"
 
-# ────────────────────────────────────────────────
-#             Gradio Oberfläche
-# ────────────────────────────────────────────────
 with gr.Blocks() as demo:
-    gr.Markdown("# Multi-Space Quick Launcher\nOmni · FireRed Edit · Ana2")
+    gr.Markdown("# Omni Image Editor with Gradio")
 
     with gr.Row():
-        space_choice = gr.Dropdown(
-            choices=list(spaces.keys()),
-            value="Omni Image Editor",
-            label="Space auswählen"
+        prompt_input = gr.Textbox(
+            label="Prompt",
+            placeholder="Describe the image you want to generate",
+            lines=2
         )
+        aspect_ratio_dropdown = gr.Dropdown(
+            label="Aspect Ratio",
+            choices=["1:1", "4:3", "3:4", "16:9", "9:16", "3:2", "2:3", "5:4", "4:5", "21:9", "9:21"],
+            value="16:9"
+        )
+    
+    generate_btn = gr.Button("Generate Image", variant="primary")
 
-    with gr.Row():
-        with gr.Column(scale=3):
-            prompt_input = gr.Textbox(
-                label="Prompt (wird nur bei Omni & FireRed genutzt)",
-                placeholder="1  oder  sexy woman nude beach  oder eigener Text",
-                lines=2
-            )
-        with gr.Column(scale=1):
-            generate_btn = gr.Button("Ausführen", variant="primary")
-
-    image_input = gr.Image(
-        label="Referenzbild (nur FireRed & Ana2 benötigt)",
-        type="filepath",
-        sources=["upload"]
-    )
-
-    output_image = gr.Image(label="Ergebnis")
-    status_text = gr.Textbox(label="Status / Fehlermeldung", interactive=False, lines=3)
-
-    # Info-Box unten
-    gr.Markdown("""
-    **Hinweise:**
-    • **Omni** → reiner Text-to-Image (16:9 fest), braucht **kein** Bild  
-    • **FireRed** → Edit-Modell, braucht fast immer ein **Startbild**  
-    • **Ana2** → wahrscheinlich NSFW/Nudity-Analyzer, braucht nur Bild, ignoriert Prompt  
-    • Bei Fehlern: API-Name oder Parameter haben sich geändert → `client.view_api()` lokal ausführen
-    """)
+    output_image = gr.Image(label="Generated Image", type="pil") # Changed type to 'pil' or 'filepath' for better handling
+    output_text = gr.Textbox(label="Status/Error", interactive=False)
 
     generate_btn.click(
-        fn=run_space,
-        inputs=[space_choice, prompt_input, image_input],
-        outputs=[output_image, status_text]
+        fn=generate_omni_image,
+        inputs=[prompt_input, aspect_ratio_dropdown],
+        outputs=[output_image, output_text]
     )
 
 demo.launch()