Update app.py

app.py

@@ -62,6 +62,7 @@ def generate_profile():
             pfp = Image.open("fallback.png").convert("RGBA")
             print("Using fallback profile picture.")
 
+        # Resize and save profile picture
         pfp = pfp.resize((pfp_size, pfp_size), Image.LANCZOS)
         pfp.save(pfp_path)  # Save before using
         print(f"Profile picture saved at {pfp_path}")
@@ -111,9 +112,11 @@ def generate_profile():
     # Send and delete image
     response = send_file(output_path, mimetype='image/png')
 
-    # Cleanup saved files
-    os.remove(output_path)
-    os.remove(pfp_path)
+    # Cleanup saved files safely
+    if os.path.exists(output_path):
+        os.remove(output_path)
+    if os.path.exists(pfp_path):
+        os.remove(pfp_path)
 
     return response