fix: use /tmp directory for file storage and serving

app.py

@@ -12,14 +12,13 @@ logging.basicConfig(level=logging.DEBUG)
 app = Flask(__name__)
 app.config['DEBUG'] = True  # This enables debug mode for Flask
 
-# Configure upload folders with absolute paths
-BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-UPLOAD_FOLDER = os.path.join(BASE_DIR, 'static', 'watermarked_images')
-THUMBNAIL_FOLDER = os.path.join(BASE_DIR, 'static', 'thumbnails')
+# Configure upload folders with absolute paths in /tmp
+UPLOAD_FOLDER = '/tmp/watermarked_images'
+THUMBNAIL_FOLDER = '/tmp/thumbnails'
 
-# Ensure directories exist
-os.makedirs(UPLOAD_FOLDER, exist_ok=True)
-os.makedirs(THUMBNAIL_FOLDER, exist_ok=True)
+# Ensure directories exist with proper permissions
+os.makedirs(UPLOAD_FOLDER, mode=0o777, exist_ok=True)
+os.makedirs(THUMBNAIL_FOLDER, mode=0o777, exist_ok=True)
 
 def get_watermark_position(img_width, img_height, wm_width, wm_height, position):
     if position == "top-left":
@@ -160,12 +159,12 @@ def upload():
         filepath = os.path.join(UPLOAD_FOLDER, filename)
         with open(filepath, 'wb') as f:
             f.write(watermarked_file.getbuffer())
-        watermarked_file_urls.append(url_for('static', filename=f'watermarked_images/{filename}'))
+        watermarked_file_urls.append(url_for('serve_watermarked', filename=filename))
         watermarked_file_paths.append(filepath)
 
         thumbnail_path = os.path.join(THUMBNAIL_FOLDER, filename)
         create_thumbnail(filepath, thumbnail_path)
-        thumbnail_urls.append(url_for('static', filename=f'thumbnails/{filename}'))
+        thumbnail_urls.append(url_for('serve_thumbnail', filename=filename))
 
     zip_filename = "watermarked_images.zip"
     zip_filepath = os.path.join(UPLOAD_FOLDER, zip_filename)
@@ -173,11 +172,24 @@ def upload():
         for file_path in watermarked_file_paths:
             zipf.write(file_path, os.path.basename(file_path))
 
-    zip_url = url_for('static', filename=f'watermarked_images/{zip_filename}')
+    zip_url = url_for('download_file', filename=zip_filename)
 
     return render_template('results.html', thumbnail_urls=thumbnail_urls, 
                            watermarked_file_urls=watermarked_file_urls, zip_url=zip_url)
 
+@app.route('/watermarked/<filename>')
+def serve_watermarked(filename):
+    return send_file(os.path.join(UPLOAD_FOLDER, filename))
+
+@app.route('/thumbnail/<filename>')
+def serve_thumbnail(filename):
+    return send_file(os.path.join(THUMBNAIL_FOLDER, filename))
+
+@app.route('/download/<filename>')
+def download_file(filename):
+    return send_file(os.path.join(UPLOAD_FOLDER, filename),
+                    as_attachment=True)
+
 if __name__ == '__main__':
     # Use the port that Hugging Face Spaces expects
     port = int(os.environ.get('PORT', 7860))