Update app.py

app.py

@@ -16,7 +16,6 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
 TEMP_DIR = "temp"
 
-
 def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
     payload = {
         "inputs": prompt,
@@ -30,14 +29,6 @@ def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
     image = Image.open(io.BytesIO(image_bytes))
     return image
 
-
-def save_image(image, prompt):
-    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-    filename = f"{prompt}-{random.randint(1, 100000)}-{timestamp}.png"
-    filepath = os.path.join(TEMP_DIR, filename)
-    image.save(filepath, format='PNG')
-    return filename
-
 def sanitize_filename(filename):
     # Replace spaces with underscores
     filename = filename.replace(" ", "_")
@@ -50,11 +41,12 @@ def sanitize_filename(filename):
 
 def save_image(image, prompt):
     timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-    filename = f"{sanitize_filename(prompt)}-{random.randint(1, 100000)}-{timestamp}.png"
+    sanitized_prompt = sanitize_filename(prompt)
+    filename = f"{sanitized_prompt}-{random.randint(1, 100000)}-{timestamp}.png"
     filepath = os.path.join(TEMP_DIR, filename)
     image.save(filepath, format='PNG')
     return filename
-    
+
 @app.route("/generate", methods=["POST"])
 def generate():
     try:
@@ -65,13 +57,13 @@ def generate():
 
         image = query(prompt, is_negative=is_negative)
         filename = save_image(image, prompt)
-            
+
         response = {
             "success": True,
-            "image": "https://mrdonstuff-dalle-3-xl-api.hf.space/temp/<filename>",
-            "filename": "<filename>"
+            "image": f"https://mrdonstuff-dalle-3-xl-api.hf.space/temp/{filename}",
+            "filename": filename
         }
-        
+
     except Exception as e:
         response = {
             "success": False,
@@ -80,12 +72,10 @@ def generate():
 
     return jsonify(response)
 
-
 @app.route("/temp/<filename>")
 def show_image(filename):
     return send_from_directory(TEMP_DIR, filename)
 
-
 if __name__ == "__main__":
     if not os.path.exists(TEMP_DIR):
         os.makedirs(TEMP_DIR)