Update app.py

app.py

@@ -4,7 +4,6 @@ import threading
 import time
 import os
 import uuid
-import tempfile
 import shutil
 
 # ---------------------------
@@ -32,7 +31,7 @@ if "prompt_history" not in st.session_state:
     st.session_state.prompt_history = []
 
 # ---------------------------
-# THEME
+# UI / THEME
 # ---------------------------
 st.set_page_config(layout="wide")
 
@@ -63,30 +62,65 @@ def save_local_copy(src_path):
 
 
 def prepare_image(input_image):
-    if hasattr(input_image, "getvalue"):
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
-        tmp.write(input_image.getvalue())
-        tmp.close()
-        return tmp.name
-    elif isinstance(input_image, str):
-        return input_image
-    return None
+    try:
+        if hasattr(input_image, "getvalue"):
+            path = os.path.join(IMAGE_DIR, f"{uuid.uuid4()}.png")
+            with open(path, "wb") as f:
+                f.write(input_image.getvalue())
+            return path
+
+        elif isinstance(input_image, str) and os.path.exists(input_image):
+            return input_image
+
+        return None
+
+    except Exception as e:
+        st.error(f"❌ Prepare Error: {e}")
+        return None
+
+
+# ---------------------------
+# 🔥 ROBUST API CALL
+# ---------------------------
+def robust_predict(func, *args, retries=3, delay=2, **kwargs):
+    for attempt in range(retries):
+        try:
+            return func(*args, **kwargs)
+
+        except Exception as e:
+            if attempt < retries - 1:
+                st.warning(f"⚠️ Retry {attempt+1}/{retries}...")
+                time.sleep(delay)
+            else:
+                st.error("❌ API dauerhaft fehlgeschlagen")
 
+                with st.expander("🔍 Debug Details"):
+                    st.code(str(e))
+
+                return None
 
 # ---------------------------
 # API FUNCTIONS
 # ---------------------------
 def generate_image(prompt):
-    return client.predict(
+    return robust_predict(
+        client.predict,
         prompt=prompt,
         api_name="/generate_image"
     )
 
 
 def edit_image(input_image, prompt):
+    if not input_image or not prompt:
+        st.warning("❗ Bild & Prompt erforderlich")
+        return None
+
     image_path = prepare_image(input_image)
+    if not image_path:
+        return None
 
-    return client.predict(
+    return robust_predict(
+        client.predict,
         input_image=handle_file(image_path),
         edit_prompt=prompt,
         api_name="/edit_image"
@@ -94,9 +128,16 @@ def edit_image(input_image, prompt):
 
 
 def process_image(input_image):
+    if not input_image:
+        st.warning("❗ Bild erforderlich")
+        return None
+
     image_path = prepare_image(input_image)
+    if not image_path:
+        return None
 
-    return client_process.predict(
+    return robust_predict(
+        client_process.predict,
         image=handle_file(image_path),
         api_name="/process_image"
     )
@@ -120,9 +161,9 @@ def show_progress(job):
     status = st.empty()
     percent = 0
 
-    messages = [
+    msgs = [
         "🔄 Processing...",
-        "🎨 Generating pixels...",
+        "🎨 Generating...",
         "🧠 AI thinking...",
         "✨ Almost done..."
     ]
@@ -131,17 +172,15 @@ def show_progress(job):
     while not job["done"]:
         percent = min(percent + 5, 95)
         bar.progress(percent)
-
-        status.text(messages[i % len(messages)])
+        status.text(msgs[i % len(msgs)])
         i += 1
-
         time.sleep(0.3)
 
     bar.progress(100)
     status.text("✅ Done!")
 
 # ---------------------------
-# NAV
+# NAVIGATION
 # ---------------------------
 page = st.radio(
     "Navigation",
@@ -158,11 +197,10 @@ st.session_state.page = page
 if page == "🖼️ Generieren":
     prompt = st.text_area("Prompt")
 
-    # 🧠 Prompt History
     if st.session_state.prompt_history:
-        st.subheader("🧠 Prompt History")
+        st.subheader("🧠 History")
         for i, p in enumerate(reversed(st.session_state.prompt_history[-5:])):
-            if st.button(p, key=f"hist{i}"):
+            if st.button(p, key=f"h{i}"):
                 prompt = p
 
     if st.button("Generieren"):
@@ -177,16 +215,12 @@ if page == "🖼️ Generieren":
         if result:
             st.image(result)
 
-            # 📥 Download
             with open(result, "rb") as f:
                 st.download_button("📥 Download", f, file_name="generated.png")
 
             path = save_local_copy(result)
             if path:
-                st.session_state.gallery.append({
-                    "path": path,
-                    "prompt": prompt
-                })
+                st.session_state.gallery.append({"path": path, "prompt": prompt})
 
 # ---------------------------
 # EDIT
@@ -214,10 +248,7 @@ elif page == "✨ Bearbeiten":
 
             path = save_local_copy(result)
             if path:
-                st.session_state.gallery.append({
-                    "path": path,
-                    "prompt": prompt
-                })
+                st.session_state.gallery.append({"path": path, "prompt": prompt})
 
 # ---------------------------
 # ANALYSE
@@ -243,10 +274,7 @@ elif page == "🧪 Analyse":
 
             if os.path.exists(result):
                 path = save_local_copy(result)
-                st.session_state.gallery.append({
-                    "path": path,
-                    "prompt": "Analyse"
-                })
+                st.session_state.gallery.append({"path": path, "prompt": "Analyse"})
 
 # ---------------------------
 # GALLERY
@@ -261,7 +289,6 @@ elif page == "📚 Galerie":
             with cols[i % 3]:
                 st.image(item["path"], caption=item["prompt"])
 
-                # 📥 Download
                 with open(item["path"], "rb") as f:
                     st.download_button("📥", f, file_name=f"img_{i}.png", key=f"d{i}")