Update app.py

app.py

@@ -16,12 +16,12 @@ client_process = Client("Galaxydude2/aaaaa")
 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
 
+if "page" not in st.session_state:
+    st.session_state.page = "🖼️ Generieren"
+
 # ---------------------------
 # Funktionen
 # ---------------------------
@@ -45,11 +45,17 @@ def generate_image(prompt):
 
 def edit_image(input_image, prompt):
     try:
+        if isinstance(input_image, str) and input_image.startswith("http"):
+            image_input = input_image
+        else:
+            image_input = handle_file(input_image)
+
         result = client_main.predict(
-            input_image=handle_file(input_image),
+            input_image=image_input,
             prompt=prompt,
             api_name="/edit_image_interface"
         )
+
         return extract_url(result[0])
     except Exception as e:
         st.error(e)
@@ -89,19 +95,21 @@ def run_with_progress(func, *args):
 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"
-])
+# 🔥 Navigation (ersetzt Tabs → erlaubt Auto-Switch)
+page = st.radio(
+    "Navigation",
+    ["🖼️ Generieren", "✨ Bearbeiten", "🧪 Process", "📚 Galerie"],
+    index=["🖼️ Generieren", "✨ Bearbeiten", "🧪 Process", "📚 Galerie"].index(st.session_state.page),
+    horizontal=True
+)
+
+st.session_state.page = page
 
 # ---------------------------
-# TAB 1: GENERATE
+# PAGE: GENERATE
 # ---------------------------
-with tab1:
+if page == "🖼️ Generieren":
     prompt = st.text_area("Bildbeschreibung")
 
     if st.button("Generieren"):
@@ -131,18 +139,33 @@ with tab1:
                 })
 
 # ---------------------------
-# TAB 2: EDIT
+# PAGE: EDIT
 # ---------------------------
-with tab2:
-    uploaded = st.file_uploader("Bild hochladen", type=["png", "jpg", "jpeg"])
+elif page == "✨ Bearbeiten":
+    st.subheader("✨ Bild bearbeiten")
+
+    use_gallery = False
+
+    # 👉 Bild aus Galerie übernommen
+    if st.session_state.selected_image:
+        st.image(st.session_state.selected_image, caption="Aus Galerie")
+        use_gallery = True
+    else:
+        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:
+        if edit_prompt:
             status = st.empty()
             progress = st.progress(0)
 
-            job = run_with_progress(edit_image, uploaded, edit_prompt)
+            if use_gallery:
+                input_image = st.session_state.selected_image
+            else:
+                input_image = uploaded
+
+            job = run_with_progress(edit_image, input_image, edit_prompt)
 
             percent = 0
             while not job["done"]:
@@ -154,20 +177,20 @@ with tab2:
             progress.progress(100)
             status.text("Fertig ✅")
 
-            img_url = job["result"]
+            result = job["result"]
 
-            if img_url:
-                st.image(img_url)
+            if result:
+                st.image(result)
                 st.session_state.gallery.append({
-                    "url": img_url,
+                    "url": result,
                     "prompt": edit_prompt
                 })
 
 # ---------------------------
-# TAB 3: PROCESS
+# PAGE: PROCESS
 # ---------------------------
-with tab3:
-    uploaded = st.file_uploader("Bild auswählen", type=["png", "jpg", "jpeg"], key="proc")
+elif page == "🧪 Process":
+    uploaded = st.file_uploader("Bild auswählen", type=["png", "jpg", "jpeg"])
 
     if st.button("Verarbeiten"):
         if uploaded:
@@ -198,9 +221,9 @@ with tab3:
                 st.write(result)
 
 # ---------------------------
-# TAB 4: GALLERY
+# PAGE: GALLERY
 # ---------------------------
-with tab4:
+elif page == "📚 Galerie":
     st.subheader("📸 Galerie")
 
     if not st.session_state.gallery:
@@ -215,52 +238,11 @@ with tab4:
             with cols[i % 3]:
                 st.image(url, caption=prompt_text)
 
+                # 🔥 AUTO SWITCH
                 if st.button("✏️ Edit", key=f"edit_{i}"):
                     st.session_state.selected_image = url
-                    st.session_state.edit_mode = True
+                    st.session_state.page = "✨ Bearbeiten"
+                    st.rerun()
 
         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
+            st.session_state.gallery = []