Upload 3 files

Dockerfile.txt

@@ -0,0 +1,10 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY app.py .
+
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,75 @@
+from flask import Flask, request, jsonify, Response
+import requests
+import os
+
+app = Flask(__name__)
+
+@app.route('/', methods=['GET'])
+def home():
+    return "Proxy API is running! Use /api/generate endpoint for image generation."
+
+@app.route('/api/generate', methods=['POST'])
+def proxy_generate():
+    try:
+        # Get the data from the request
+        data = request.json
+        
+        # Forward the request to the actual API
+        response = requests.post(
+            "https://aiart-zroo.onrender.com/api/generate",
+            headers={"Content-Type": "application/json"},
+            json=data,
+            timeout=60  # Set a reasonable timeout
+        )
+        
+        # If successful, modify the response to use our proxy for the image
+        if response.status_code == 200:
+            try:
+                result = response.json()
+                if 'image_url' in result:
+                    # Get the original image filename
+                    original_url = result['image_url']
+                    filename = original_url.split('/')[-1]
+                    
+                    # Replace the image URL with our proxy URL
+                    proxy_url = f"/proxy-image/{filename}"
+                    result['original_image_url'] = original_url
+                    result['image_url'] = proxy_url
+                    
+                    return jsonify(result)
+            except:
+                pass
+        
+        # Return the response from the API
+        return response.content, response.status_code, {'Content-Type': response.headers.get('Content-Type', 'application/json')}
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+# Add an image proxy endpoint
+@app.route('/proxy-image/<filename>', methods=['GET'])
+def proxy_image(filename):
+    try:
+        # Construct the original URL
+        original_url = f"https://aiart-zroo.onrender.com/images/{filename}"
+        
+        # Get the image with stream=True to handle binary data properly
+        response = requests.get(original_url, stream=True)
+        
+        # Return the image data with explicit content type
+        return Response(
+            response.raw.read(),  # Use raw.read() for binary data
+            status=response.status_code,
+            content_type='image/png',  # Force content type to image/png
+            headers={
+                'Cache-Control': 'no-cache',
+                'Content-Disposition': f'inline; filename={filename}'
+            }
+        )
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+#new
+if __name__ == '__main__':
+    # Get port from environment variable or use 7860 as default
+    port = int(os.environ.get('PORT', 7860))
+    app.run(host='0.0.0.0', port=port)

requirements (1).txt

@@ -0,0 +1,5 @@
+flask==2.0.1
+werkzeug==2.0.3
+requests==2.28.1
+gunicorn==20.1.0
+