Update app.py

app.py

@@ -11,11 +11,15 @@ from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 from diffusers import FluxPipeline
 
-# S3 Configuration
-S3_BUCKET = "afri"
-S3_REGION = "eu-west-3"
-S3_ACCESS_KEY_ID = "AKIAQQABC7IQWFLKSE62"
-S3_SECRET_ACCESS_KEY = "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"
+# S3 Configuration from environment variables
+S3_BUCKET = os.getenv("S3_BUCKET")
+S3_REGION = os.getenv("S3_REGION")
+S3_ACCESS_KEY_ID = os.getenv("S3_ACCESS_KEY_ID")
+S3_SECRET_ACCESS_KEY = os.getenv("S3_SECRET_ACCESS_KEY")
+
+# Validate S3 environment variables
+if not all([S3_BUCKET, S3_REGION, S3_ACCESS_KEY_ID, S3_SECRET_ACCESS_KEY]):
+    raise ValueError("Missing required S3 environment variables")
 
 # Set up S3 client
 s3_client = boto3.client('s3', 
@@ -28,7 +32,6 @@ cache_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
 os.environ["TRANSFORMERS_CACHE"] = cache_path
 os.environ["HF_HUB_CACHE"] = cache_path
 os.environ["HF_HOME"] = cache_path
-
 if not os.path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
@@ -59,11 +62,11 @@ class InferenceRequest(BaseModel):
 class Timer:
     def __init__(self, method_name="timed process"):
         self.method = method_name
-
+    
     def __enter__(self):
         self.start = time.time()
         print(f"{self.method} starts")
-
+    
     def __exit__(self, exc_type, exc_val, exc_tb):
         end = time.time()
         print(f"{self.method} took {str(round(end - self.start, 2))}s")