Create app.py

app.py

@@ -0,0 +1,266 @@
+import streamlit as st
+from gradio_client import Client, handle_file
+import re
+import threading
+import time
+
+# ---------------------------
+# Clients
+# ---------------------------
+client_main = Client("selfit-camera/Omni-Image-Editor")
+client_process = Client("Galaxydude2/aaaaa")
+
+# ---------------------------
+# Session State
+# ---------------------------
+if "gallery" not in st.session_state:
+    st.session_state.gallery = []
+
+if "edit_mode" not in st.session_state:
+    st.session_state.edit_mode = False
+
+if "selected_image" not in st.session_state:
+    st.session_state.selected_image = None
+
+# ---------------------------
+# Funktionen
+# ---------------------------
+def extract_url(html):
+    match = re.search(r"src='([^']+)'", html)
+    return match.group(1) if match else None
+
+
+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(e)
+        return None
+
+
+def edit_image(input_image, prompt):
+    try:
+        result = client_main.predict(
+            input_image=handle_file(input_image),
+            prompt=prompt,
+            api_name="/edit_image_interface"
+        )
+        return extract_url(result[0])
+    except Exception as e:
+        st.error(e)
+        return None
+
+
+def process_image(input_image):
+    try:
+        result = client_process.predict(
+            image=handle_file(input_image),
+            api_name="/process_image"
+        )
+        return result
+    except Exception as e:
+        st.error(e)
+        return None
+
+
+# ---------------------------
+# Async Helper
+# ---------------------------
+def run_with_progress(func, *args):
+    result_container = {"result": None, "done": False}
+
+    def task():
+        result_container["result"] = func(*args)
+        result_container["done"] = True
+
+    thread = threading.Thread(target=task)
+    thread.start()
+    return result_container
+
+
+# ---------------------------
+# UI Setup
+# ---------------------------
+st.set_page_config(page_title="Omni Image Studio", layout="wide")
+
+st.title("🎨 Omni Image Studio")
+st.write("Generieren • Bearbeiten • Verarbeiten • Galerie")
+
+tab1, tab2, tab3, tab4 = st.tabs([
+    "🖼️ Generieren",
+    "✨ Bearbeiten",
+    "🧪 Process",
+    "📚 Galerie"
+])
+
+# ---------------------------
+# TAB 1: GENERATE
+# ---------------------------
+with tab1:
+    prompt = st.text_area("Bildbeschreibung")
+
+    if st.button("Generieren"):
+        if prompt:
+            status = st.empty()
+            progress = st.progress(0)
+
+            job = run_with_progress(generate_image, prompt)
+
+            percent = 0
+            while not job["done"]:
+                percent = min(percent + 5, 95)
+                progress.progress(percent)
+                status.text(f"Generiere Bild... {percent}%")
+                time.sleep(0.2)
+
+            progress.progress(100)
+            status.text("Fertig ✅")
+
+            img_url = job["result"]
+
+            if img_url:
+                st.image(img_url)
+                st.session_state.gallery.append({
+                    "url": img_url,
+                    "prompt": prompt
+                })
+
+# ---------------------------
+# TAB 2: EDIT
+# ---------------------------
+with tab2:
+    uploaded = st.file_uploader("Bild hochladen", type=["png", "jpg", "jpeg"])
+    edit_prompt = st.text_area("Bearbeitung")
+
+    if st.button("Bearbeiten"):
+        if uploaded and edit_prompt:
+            status = st.empty()
+            progress = st.progress(0)
+
+            job = run_with_progress(edit_image, uploaded, edit_prompt)
+
+            percent = 0
+            while not job["done"]:
+                percent = min(percent + 5, 95)
+                progress.progress(percent)
+                status.text(f"Bearbeite Bild... {percent}%")
+                time.sleep(0.2)
+
+            progress.progress(100)
+            status.text("Fertig ✅")
+
+            img_url = job["result"]
+
+            if img_url:
+                st.image(img_url)
+                st.session_state.gallery.append({
+                    "url": img_url,
+                    "prompt": edit_prompt
+                })
+
+# ---------------------------
+# TAB 3: PROCESS
+# ---------------------------
+with tab3:
+    uploaded = st.file_uploader("Bild auswählen", type=["png", "jpg", "jpeg"], key="proc")
+
+    if st.button("Verarbeiten"):
+        if uploaded:
+            status = st.empty()
+            progress = st.progress(0)
+
+            job = run_with_progress(process_image, uploaded)
+
+            percent = 0
+            while not job["done"]:
+                percent = min(percent + 5, 95)
+                progress.progress(percent)
+                status.text(f"Verarbeite Bild... {percent}%")
+                time.sleep(0.2)
+
+            progress.progress(100)
+            status.text("Fertig ✅")
+
+            result = job["result"]
+
+            if isinstance(result, str) and result.startswith("http"):
+                st.image(result)
+                st.session_state.gallery.append({
+                    "url": result,
+                    "prompt": "Processed Image"
+                })
+            else:
+                st.write(result)
+
+# ---------------------------
+# TAB 4: GALLERY
+# ---------------------------
+with tab4:
+    st.subheader("📸 Galerie")
+
+    if not st.session_state.gallery:
+        st.info("Keine Bilder vorhanden")
+    else:
+        cols = st.columns(3)
+
+        for i, item in enumerate(st.session_state.gallery):
+            url = item.get("url")
+            prompt_text = item.get("prompt", "Kein Prompt")
+
+            with cols[i % 3]:
+                st.image(url, caption=prompt_text)
+
+                if st.button("✏️ Edit", key=f"edit_{i}"):
+                    st.session_state.selected_image = url
+                    st.session_state.edit_mode = True
+
+        if st.button("🗑️ Galerie leeren"):
+            st.session_state.gallery = []
+
+# ---------------------------
+# GLOBAL EDIT (aus Galerie)
+# ---------------------------
+if st.session_state.edit_mode:
+    st.divider()
+    st.subheader("✏️ Bild aus Galerie bearbeiten")
+
+    st.image(st.session_state.selected_image)
+
+    edit_prompt_gallery = st.text_area("Bearbeitung", key="gallery_edit")
+
+    if st.button("🚀 Anwenden"):
+        status = st.empty()
+        progress = st.progress(0)
+
+        job = run_with_progress(
+            edit_image,
+            st.session_state.selected_image,
+            edit_prompt_gallery
+        )
+
+        percent = 0
+        while not job["done"]:
+            percent = min(percent + 5, 95)
+            progress.progress(percent)
+            status.text(f"Bearbeite Bild... {percent}%")
+            time.sleep(0.2)
+
+        progress.progress(100)
+        status.text("Fertig ✅")
+
+        result = job["result"]
+
+        if result:
+            st.image(result)
+            st.session_state.gallery.append({
+                "url": result,
+                "prompt": edit_prompt_gallery
+            })
+
+    if st.button("❌ Abbrechen"):
+        st.session_state.edit_mode = False