Update app.py

app.py

@@ -4,6 +4,7 @@ from io import BytesIO
 from PIL import Image
 import os
 import tempfile
+import mimetypes
 
 TOKEN = os.getenv("TOKEN1")
 API_URL = os.getenv("API_URL")
@@ -69,9 +70,16 @@ def download_image(image_url):
     try:
         response = requests.get(image_url, stream=True)
         response.raise_for_status()
-        
-        # Create a temporary file
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
+
+        # Get the content type from the headers
+        content_type = response.headers.get('content-type')
+        extension = mimetypes.guess_extension(content_type)
+
+        if not extension:
+             extension = ".png" # Default to .png if can't determine the extension
+
+        # Create a temporary file with the correct extension
+        with tempfile.NamedTemporaryFile(suffix=extension, delete=False) as tmp_file:
             for chunk in response.iter_content(chunk_size=8192):
                 tmp_file.write(chunk)
             temp_file_path = tmp_file.name