Update app.py

app.py

@@ -3,6 +3,8 @@ from gradio_client import Client, handle_file
 import re
 import threading
 import time
+import tempfile
+import requests
 
 # ---------------------------
 # Clients
@@ -23,13 +25,33 @@ if "page" not in st.session_state:
     st.session_state.page = "🖼️ Generieren"
 
 # ---------------------------
-# Funktionen
+# Helper
 # ---------------------------
 def extract_url(html):
     match = re.search(r"src='([^']+)'", html)
     return match.group(1) if match else None
 
 
+def url_to_file(url):
+    try:
+        r = requests.get(url)
+        if r.status_code != 200:
+            st.error(f"❌ Download Error: HTTP {r.status_code}")
+            return None
+
+        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        tmp.write(r.content)
+        tmp.close()
+        return tmp.name
+
+    except Exception as e:
+        st.error(f"❌ URL Download Exception: {e}")
+        return None
+
+
+# ---------------------------
+# API FUNCTIONS
+# ---------------------------
 def generate_image(prompt):
     try:
         result = client_main.predict(
@@ -38,15 +60,19 @@ def generate_image(prompt):
             api_name="/text_to_image_interface"
         )
         return extract_url(result[0])
+
     except Exception as e:
-        st.error(e)
+        st.error(f"❌ Generate Error: {e}")
         return None
 
 
 def edit_image(input_image, prompt):
     try:
         if isinstance(input_image, str) and input_image.startswith("http"):
-            image_input = input_image
+            file_path = url_to_file(input_image)
+            if not file_path:
+                return None
+            image_input = handle_file(file_path)
         else:
             image_input = handle_file(input_image)
 
@@ -57,20 +83,31 @@ def edit_image(input_image, prompt):
         )
 
         return extract_url(result[0])
+
     except Exception as e:
-        st.error(e)
+        st.error(f"❌ Edit Error: {e}")
         return None
 
 
 def process_image(input_image):
     try:
+        if isinstance(input_image, str) and input_image.startswith("http"):
+            file_path = url_to_file(input_image)
+            if not file_path:
+                return None
+            image_input = handle_file(file_path)
+        else:
+            image_input = handle_file(input_image)
+
         result = client_process.predict(
-            image=handle_file(input_image),
+            image=image_input,
             api_name="/process_image"
         )
+
         return result
+
     except Exception as e:
-        st.error(e)
+        st.error(f"❌ Process Error: {e}")
         return None
 
 
@@ -84,30 +121,29 @@ def run_with_progress(func, *args):
         result_container["result"] = func(*args)
         result_container["done"] = True
 
-    thread = threading.Thread(target=task)
-    thread.start()
+    threading.Thread(target=task).start()
     return result_container
 
 
 # ---------------------------
-# UI Setup
+# UI
 # ---------------------------
 st.set_page_config(page_title="Omni Image Studio", layout="wide")
 
 st.title("🎨 Omni Image Studio")
 
-# 🔥 Navigation (ersetzt Tabs → erlaubt Auto-Switch)
+# Navigation
 page = st.radio(
     "Navigation",
-    ["🖼️ Generieren", "✨ Bearbeiten", "🧪 Process", "📚 Galerie"],
-    index=["🖼️ Generieren", "✨ Bearbeiten", "🧪 Process", "📚 Galerie"].index(st.session_state.page),
+    ["🖼️ Generieren", "✨ Bearbeiten", "🧪 Analyse", "📚 Galerie"],
+    index=["🖼️ Generieren", "✨ Bearbeiten", "🧪 Analyse", "📚 Galerie"].index(st.session_state.page),
     horizontal=True
 )
 
 st.session_state.page = page
 
 # ---------------------------
-# PAGE: GENERATE
+# GENERATE
 # ---------------------------
 if page == "🖼️ Generieren":
     prompt = st.text_area("Bildbeschreibung")
@@ -139,32 +175,25 @@ if page == "🖼️ Generieren":
                 })
 
 # ---------------------------
-# PAGE: EDIT
+# EDIT
 # ---------------------------
 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
+        input_image = st.session_state.selected_image
     else:
-        uploaded = st.file_uploader("Bild hochladen", type=["png", "jpg", "jpeg"])
+        uploaded = st.file_uploader("Bild hochladen", type=["png", "jpg"])
+        input_image = uploaded
 
     edit_prompt = st.text_area("Bearbeitung")
 
     if st.button("Bearbeiten"):
-        if edit_prompt:
+        if input_image and edit_prompt:
             status = st.empty()
             progress = st.progress(0)
 
-            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
@@ -187,23 +216,30 @@ elif page == "✨ Bearbeiten":
                 })
 
 # ---------------------------
-# PAGE: PROCESS
+# ANALYSE (PROCESS)
 # ---------------------------
-elif page == "🧪 Process":
-    uploaded = st.file_uploader("Bild auswählen", type=["png", "jpg", "jpeg"])
+elif page == "🧪 Analyse":
+    st.subheader("🧪 Bild Analyse")
+
+    if st.session_state.selected_image:
+        st.image(st.session_state.selected_image, caption="Aus Galerie")
+        input_image = st.session_state.selected_image
+    else:
+        uploaded = st.file_uploader("Bild auswählen", type=["png", "jpg"])
+        input_image = uploaded
 
-    if st.button("Verarbeiten"):
-        if uploaded:
+    if st.button("Analysieren"):
+        if input_image:
             status = st.empty()
             progress = st.progress(0)
 
-            job = run_with_progress(process_image, uploaded)
+            job = run_with_progress(process_image, input_image)
 
             percent = 0
             while not job["done"]:
                 percent = min(percent + 5, 95)
                 progress.progress(percent)
-                status.text(f"Verarbeite Bild... {percent}%")
+                status.text(f"Analysiere Bild... {percent}%")
                 time.sleep(0.2)
 
             progress.progress(100)
@@ -215,13 +251,13 @@ elif page == "🧪 Process":
                 st.image(result)
                 st.session_state.gallery.append({
                     "url": result,
-                    "prompt": "Processed Image"
+                    "prompt": "Analyse Ergebnis"
                 })
             else:
                 st.write(result)
 
 # ---------------------------
-# PAGE: GALLERY
+# GALLERY
 # ---------------------------
 elif page == "📚 Galerie":
     st.subheader("📸 Galerie")
@@ -232,17 +268,27 @@ elif page == "📚 Galerie":
         cols = st.columns(3)
 
         for i, item in enumerate(st.session_state.gallery):
-            url = item.get("url")
-            prompt_text = item.get("prompt", "Kein Prompt")
+            url = item["url"]
+            prompt_text = item["prompt"]
 
             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.page = "✨ Bearbeiten"
-                    st.rerun()
+                col1, col2 = st.columns(2)
+
+                # 🔥 EDIT BUTTON
+                with col1:
+                    if st.button("✏️ Edit", key=f"edit_{i}"):
+                        st.session_state.selected_image = url
+                        st.session_state.page = "✨ Bearbeiten"
+                        st.rerun()
+
+                # 🔥 ANALYSE BUTTON
+                with col2:
+                    if st.button("🧪 Analyse", key=f"proc_{i}"):
+                        st.session_state.selected_image = url
+                        st.session_state.page = "🧪 Analyse"
+                        st.rerun()
 
         if st.button("🗑️ Galerie leeren"):
             st.session_state.gallery = []