fix: response to s3 images

server.py

@@ -13,6 +13,7 @@ import requests
 import uuid
 import importlib
 import io
+import boto3
 from flask import Flask, request, jsonify, send_file
 import numpy as np
 
@@ -228,8 +229,8 @@ def create_app() -> Flask:
             print(f"Error in /generate/{visa_type}: {e}")
             return jsonify({"error": str(e)}), 500
         
-    @app.route("/generate-passport-photo", methods=["POST"])
-    def generate_passport():
+    @app.route("/generate-visa-photo", methods=["POST"])
+    def generate_passport_visa():
         data = request.get_json()
         bg_color_name = data.get('bg_color_name', 'white')
         output_size = data.get('output_size', (600, 800))
@@ -276,13 +277,43 @@ def create_app() -> Flask:
                 bottom_margin=1
             )
             # Return result
+            # Encode image to memory buffer
             is_success, buffer = cv2.imencode(".jpg", final_passport)
-            return send_file(io.BytesIO(buffer), mimetype='image/jpeg')
+            if not is_success:
+                 return jsonify({"error": "Failed to encode image"}), 500
+            
+            # Prepare for S3 Upload
+            file_stream = io.BytesIO(buffer)
+            s3_client = boto3.client('s3',
+                aws_access_key_id=os.getenv("AWS_ACCESS_KEY"),
+                aws_secret_access_key=os.getenv("AWS_SECRET_KEY"),
+                region_name=os.getenv("AWS_REGION")          
+            )
+            
+            # You should set this env var or hardcode your bucket
+            S3_BUCKET_NAME = os.getenv("S3_PUBLIC_BUCKET", "spun-core-api-temp-development") 
+            file_key = f"generated-visa-photo/{uuid.uuid4()}.jpg"
+            
+            # Upload
+            s3_client.upload_fileobj(
+                file_stream,
+                S3_BUCKET_NAME,
+                file_key,
+                ExtraArgs={'ContentType': 'image/jpeg', 'ACL': 'public-read'} # ACL optional depending on bucket settings
+            )
+            
+            # Construct URL
+            # Note: This assumes standard AWS S3 URL format. Adjust if using CloudFront.
+            s3_url = f"https://{S3_BUCKET_NAME}.s3.amazonaws.com/{file_key}"
+            
+            return jsonify({
+                "status": "success",
+                "url": s3_url
+            })
 
         except Exception as e:
             print(f"Passport Error: {e}")
             return jsonify({"error": str(e)}), 500
-
     @app.route('/', methods=['GET'])
     def hello_world():
         return "Flask server is running.", 200