feat: passport image gen

.env.example

@@ -1,2 +0,0 @@
-BASE_URI=ttps://spunteam-api-web-crawler.hf.space
-URL_CE_BOT=https://echo.spun.global

requirements.txt

@@ -4,4 +4,5 @@ requests
 python-dotenv
 streamlit
 pydantic
-beautifulsoup4
+beautifulsoup4
+boto3

src/pages/05_Passport_Image_Generator.py

@@ -0,0 +1,189 @@
+import streamlit as st
+import requests
+import os
+import boto3
+import uuid
+from dotenv import load_dotenv
+from PIL import Image
+
+# --- 1. CONFIGURATION ---
+st.set_page_config(page_title="Passport Photo Generator", layout="wide", page_icon="📸")
+load_dotenv()
+
+# Load Config from .env
+API_BASE_URL = os.getenv("API_BASE_URL", "https://spunteam-api-web-crawler.hf.space")
+AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY_ID")
+AWS_SECRET_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
+AWS_REGION = os.getenv("AWS_REGION", "ap-southeast-1")
+S3_BUCKET_NAME = os.getenv("S3_BUCKET_NAME")
+
+# --- 2. S3 HELPER FUNCTIONS ---
+def get_s3_client():
+    if not AWS_ACCESS_KEY or not AWS_SECRET_KEY:
+        st.error("❌ AWS Credentials missing in .env file.")
+        return None
+    return boto3.client(
+        's3',
+        aws_access_key_id=AWS_ACCESS_KEY,
+        aws_secret_access_key=AWS_SECRET_KEY,
+        region_name=AWS_REGION
+    )
+import logging
+import boto3
+from botocore.exceptions import NoCredentialsError, ClientError
+from typing import Optional, IO
+
+# Ensure your s3_client is initialized globally or passed in
+s3_client = boto3.client('s3')
+
+def upload_buffer_and_get_presigned_url(
+    file_obj: IO, 
+    bucket: str, 
+    s3_key: str, 
+    content_type: str = 'image/jpeg'
+) -> Optional[str]:
+    """
+    Uploads an in-memory file object to S3 and returns its presigned URL.
+    """
+    try:
+        logging.info(f"Uploading data to s3://{bucket}/{s3_key}...")
+        
+        # Reset file pointer to the beginning to ensure full upload
+        file_obj.seek(0)
+        
+        # Use upload_fileobj for memory streams (BytesIO, SpooledTemporaryFile)
+        s3_client.upload_fileobj(
+            file_obj, 
+            bucket, 
+            s3_key,
+            ExtraArgs={'ContentType': content_type} # Crucial for browser preview
+        )
+        
+        logging.info("Generating presigned URL for access...")
+        
+        # Generate a presigned URL valid for 1 hour (3600 seconds)
+        presigned_url = s3_client.generate_presigned_url(
+            'get_object',
+            Params={'Bucket': bucket, 'Key': s3_key},
+            ExpiresIn=3600 
+        )
+        
+        logging.info("Upload and signing successful.")
+        return presigned_url
+
+    except NoCredentialsError:
+        logging.error("AWS credentials not found. Configure them to upload to S3.")
+        return None
+    except ClientError as e:
+        logging.error(f"S3 Client Error: {e}")
+        return None
+    except Exception as e:
+        logging.error(f"Unexpected error during S3 upload: {e}")
+        return None
+    
+# --- 3. MAIN UI ---
+st.title("📸 AI Passport & Visa Photo Generator")
+st.markdown("Upload a raw photo, choose a background, and get a compliant passport photo.")
+
+# Sidebar for Settings
+with st.sidebar:
+    st.header("Settings")
+    bg_color = st.selectbox(
+        "Select Background Color",
+        ("white", "id_red", "id_blue", "light_blue", "off_white"),
+        index=0,
+        help="Standard passports use White. Indonesia uses Red/Blue."
+    )
+    st.info(f"Target API: `{API_BASE_URL}`")
+    st.info(f"S3 Bucket: `{S3_BUCKET_NAME}`")
+
+# Main Input Area
+uploaded_file = st.file_uploader("Upload your photo (JPG/PNG)", type=["jpg", "jpeg", "png"])
+
+# --- 4. LAYOUT: INPUT VS RESPONSE ---
+col1, col2 = st.columns(2)
+
+# Global variables to hold state
+input_url = None
+
+with col1:
+    st.subheader("1. Input Image")
+    if uploaded_file:
+        # Display the uploaded image directly from memory
+        st.image(uploaded_file, caption="Original Photo", use_column_width=True)
+    else:
+        st.info("Waiting for upload...")
+
+with col2:
+    st.subheader("2. Response Image")
+    
+    # Only show button if file is uploaded
+    if uploaded_file:
+        generate_btn = st.button("🚀 Process Photo", type="primary", use_container_width=True)
+        
+        if generate_btn:
+            # A. UPLOAD TO S3
+            with st.status("Uploading input to S3...", expanded=True) as status:
+                # Reset file pointer to beginning before upload
+                uploaded_file.seek(0)
+                input_url = upload_buffer_and_get_presigned_url(
+                file_obj=uploaded_file,
+                bucket=S3_BUCKET_NAME,
+                s3_key=f"inputs/{uploaded_file.name}",
+                content_type=uploaded_file.type
+            )
+                
+                if input_url:
+                    status.write(f"✅ Uploaded to: {input_url}")
+                    
+                    # B. CALL API
+                    status.write("🤖 Sending to AI Engine...")
+                    
+                    api_url = f"{API_BASE_URL}/generate-passport-photo"
+                    payload = {
+                        "raw_photo": input_url,
+                        "bg_color_name": bg_color
+                    }
+                    
+                    try:
+                        response = requests.post(api_url, json=payload, timeout=60)
+                        
+                        if response.status_code == 200:
+                            image_bytes = response.content
+                            
+                            status.update(label="✅ Processing Complete!", state="complete", expanded=False)
+                            
+                            # 2. Display Result (st.image can read bytes directly)
+                            st.success("Generation Successful!")
+                            st.image(image_bytes, caption=f"Processed ({bg_color})", use_container_width=True)
+                            
+                            # 3. Download Button (Native Streamlit button for bytes)
+                            st.download_button(
+                                label="⬇️ Download HD Image",
+                                data=image_bytes,
+                                file_name="passport_result.jpg",
+                                mime="image/jpeg",
+                                type="primary"
+                            )
+
+                        else:
+                            status.update(label="❌ API Error", state="error")
+                            # Try to decode error message if it's JSON, otherwise show text
+                            try:
+                                err_msg = response.json().get("error", response.text)
+                            except:
+                                err_msg = response.text
+                            st.error(f"API Failed: {err_msg}")
+                            
+                    except Exception as e:
+                        status.update(label="❌ Connection Error", state="error")
+                        st.error(f"Could not connect to API: {e}")
+                else:
+                    status.update(label="❌ Upload Failed", state="error")
+    else:
+        st.info("Result will appear here...")
+
+# --- 5. DEBUG SECTION (Optional) ---
+# with st.expander("Debug Info"):
+#     if 'data' in locals():
+#         st.json(data)