Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 import requests
+import os
+import base64
+
 
 api_url = "https://codinglehrer.pythonanywhere.com/generate_image"
 
@@ -14,7 +17,18 @@ def generate_image(prompt, seed=0, negative_prompt="", model="Deliberate", sampl
     }
     response = requests.post(api_url, data=data)
     if response.status_code == 200:
-        return response.json()["url"]
+        # Decode the base64 image string
+        img_bytes = base64.b64decode(response.json()["url"])
+        
+        # Save the image in the images folder
+        if not os.path.exists("images"):
+            os.makedirs("images")
+        filename = f"images/{seed}.png"
+        with open(filename, 'wb') as f:
+            f.write(img_bytes)
+        
+        # Return the path of the saved image file
+        return filename
     else:
         return None