Delete app.py

app.py

@@ -1,284 +0,0 @@
-import streamlit as st
-from gradio_client import Client, handle_file
-import re
-import threading
-import time
-import requests
-import os
-import uuid
-import tempfile
-
-# ---------------------------
-# CONFIG
-# ---------------------------
-IMAGE_DIR = "data/images"
-os.makedirs(IMAGE_DIR, exist_ok=True)
-
-client_main = Client("selfit-camera/Omni-Image-Editor")
-client_process = Client("SLSLK/n8")
-
-# ---------------------------
-# SESSION
-# ---------------------------
-if "gallery" not in st.session_state:
-    st.session_state.gallery = []
-
-if "selected_image" not in st.session_state:
-    st.session_state.selected_image = None
-
-if "page" not in st.session_state:
-    st.session_state.page = "🖼️ Generieren"
-
-# ---------------------------
-# HELPERS
-# ---------------------------
-def extract_url(html):
-    match = re.search(r"src='([^']+)'", html)
-    return match.group(1) if match else None
-
-
-def save_image_from_url(url):
-    try:
-        r = requests.get(url)
-        if r.status_code != 200:
-            st.error(f"❌ Download Error HTTP {r.status_code}")
-            return None
-
-        path = os.path.join(IMAGE_DIR, f"{uuid.uuid4()}.png")
-        with open(path, "wb") as f:
-            f.write(r.content)
-
-        return path
-    except Exception as e:
-        st.error(f"❌ Save Error: {e}")
-        return None
-
-
-def prepare_image_for_api(input_image):
-    """
-    🔥 UNIVERSAL FIX:
-    - Upload → temp file
-    - Path → direkt
-    - URL → direkt
-    → immer handle_file(...)
-    """
-    try:
-        # 📥 Upload (Streamlit)
-        if hasattr(input_image, "getvalue"):
-            tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-            tmp.write(input_image.getvalue())
-            tmp.close()
-            return handle_file(tmp.name)
-
-        # 📁 Lokaler Pfad oder URL
-        elif isinstance(input_image, str):
-            return handle_file(input_image)
-
-        else:
-            return None
-
-    except Exception as e:
-        st.error(f"❌ Prepare Error: {e}")
-        return None
-
-
-# ---------------------------
-# API FUNCTIONS
-# ---------------------------
-def generate_image(prompt):
-    try:
-        result = client_main.predict(
-            prompt=prompt,
-            aspect_ratio="16:9",
-            api_name="/text_to_image_interface"
-        )
-        return extract_url(result[0])
-    except Exception as e:
-        st.error(f"❌ Generate Error: {e}")
-        return None
-
-
-def edit_image(input_image, prompt):
-    try:
-        image_input = prepare_image_for_api(input_image)
-        if not image_input:
-            st.error("❌ Invalid image input")
-            return None
-
-        result = client_main.predict(
-            input_image=image_input,
-            prompt=prompt,
-            api_name="/edit_image_interface"
-        )
-
-        html = result[0]
-        return extract_url(html)
-
-    except Exception as e:
-        st.error(f"❌ Edit Error: {e}")
-        return None
-
-
-def process_image(input_image):
-    try:
-        image_input = prepare_image_for_api(input_image)
-        if not image_input:
-            st.error("❌ Invalid image input")
-            return None
-
-        result = client_process.predict(
-            image=image_input,
-            api_name="/process_image"
-        )
-
-        return result
-
-    except Exception as e:
-        st.error(f"❌ Process Error: {e}")
-        return None
-
-
-# ---------------------------
-# ASYNC
-# ---------------------------
-def run_async(func, *args):
-    result = {"done": False, "data": None}
-
-    def task():
-        result["data"] = func(*args)
-        result["done"] = True
-
-    threading.Thread(target=task).start()
-    return result
-
-
-def show_progress(job):
-    progress_bar = st.progress(0)
-    percent = 0
-
-    while not job["done"]:
-        percent += 5
-        if percent > 95:
-            percent = 95
-
-        progress_bar.progress(percent)
-        time.sleep(0.2)
-
-    progress_bar.progress(100)
-
-
-# ---------------------------
-# UI
-# ---------------------------
-st.set_page_config(layout="wide")
-st.title("🎨 Omni Image Studio")
-
-page = st.radio(
-    "Navigation",
-    ["🖼️ Generieren", "✨ Bearbeiten", "🧪 Analyse", "📚 Galerie"],
-    index=["🖼️ Generieren", "✨ Bearbeiten", "🧪 Analyse", "📚 Galerie"].index(st.session_state.page),
-    horizontal=True
-)
-
-st.session_state.page = page
-
-# ---------------------------
-# GENERATE
-# ---------------------------
-if page == "🖼️ Generieren":
-    prompt = st.text_area("Prompt")
-
-    if st.button("Generieren"):
-        job = run_async(generate_image, prompt)
-        show_progress(job)
-
-        url = job["data"]
-        if url:
-            path = save_image_from_url(url)
-            st.image(path)
-
-            st.session_state.gallery.append({
-                "path": path,
-                "prompt": prompt
-            })
-
-# ---------------------------
-# EDIT
-# ---------------------------
-elif page == "✨ Bearbeiten":
-    if st.session_state.selected_image:
-        st.image(st.session_state.selected_image)
-        input_image = st.session_state.selected_image
-    else:
-        input_image = st.file_uploader("Upload Image")
-
-    prompt = st.text_area("Edit Prompt")
-
-    if st.button("Bearbeiten"):
-        job = run_async(edit_image, input_image, prompt)
-        show_progress(job)
-
-        url = job["data"]
-        if url:
-            path = save_image_from_url(url)
-            st.image(path)
-
-            st.session_state.gallery.append({
-                "path": path,
-                "prompt": prompt
-            })
-
-# ---------------------------
-# ANALYSE
-# ---------------------------
-elif page == "🧪 Analyse":
-    if st.session_state.selected_image:
-        st.image(st.session_state.selected_image)
-        input_image = st.session_state.selected_image
-    else:
-        input_image = st.file_uploader("Upload Image")
-
-    if st.button("Analysieren"):
-        job = run_async(process_image, input_image)
-        show_progress(job)
-
-        result = job["data"]
-
-        if isinstance(result, str) and result.startswith("http"):
-            path = save_image_from_url(result)
-            st.image(path)
-
-            st.session_state.gallery.append({
-                "path": path,
-                "prompt": "Analyse"
-            })
-        else:
-            st.write(result)
-
-# ---------------------------
-# GALLERY
-# ---------------------------
-elif page == "📚 Galerie":
-    if not st.session_state.gallery:
-        st.info("Keine Bilder")
-    else:
-        cols = st.columns(3)
-
-        for i, item in enumerate(st.session_state.gallery):
-            with cols[i % 3]:
-                st.image(item["path"], caption=item["prompt"])
-
-                c1, c2 = st.columns(2)
-
-                if c1.button("✏️ Edit", key=f"e{i}"):
-                    st.session_state.selected_image = item["path"]
-                    st.session_state.page = "✨ Bearbeiten"
-                    st.rerun()
-
-                if c2.button("🧪 Analyse", key=f"a{i}"):
-                    st.session_state.selected_image = item["path"]
-                    st.session_state.page = "🧪 Analyse"
-                    st.rerun()
-
-        if st.button("🗑️ Clear"):
-            st.session_state.gallery = []