Update sd.py

sd.py

@@ -17,7 +17,7 @@ def load_config():
     with open('config.yaml', 'r') as f:
         return yaml.safe_load(f)
 
-def upload_to_imgbb(image_data):
+def upload_to_imgbb(image_data, file_name):
     """
     Uploads image data to ImgBB and returns the URL.
 
@@ -32,10 +32,9 @@ def upload_to_imgbb(image_data):
             "https://api.imgbb.com/1/upload",
             params={
                 "key": IMGBB_API_KEY,
-                "expiration": 600  # Optional: Set expiration time in seconds
             },
             files={
-                "image": ("image.jpg", image_data)
+                "image": (file_name, image_data)
             }
         )
         response.raise_for_status()
@@ -100,7 +99,7 @@ def generate_sd(prompt):
                 f.write(image_data)
 
             # Upload image to ImgBB
-            image_url = upload_to_imgbb(image_data)
+            image_url = upload_to_imgbb(image_data, image_filename)
 
             # Convert the path to use forward slashes
             image_path = image_path.replace('\\', '/')