Update app.py

app.py

@@ -14,13 +14,12 @@ import uuid
 
 load_dotenv()
 
-ENDPOINT_URL = os.getenv("ENDPOINT_URL")
 AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
 AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
-BUCKET_NAME = "techTailor"
+BUCKET_NAME = "tech-tailor"
 s3_client = boto3.client(
     "s3",
-    endpoint_url = ENDPOINT_URL
+    region_name='ap-south-1',
     aws_access_key_id=AWS_ACCESS_KEY_ID, 
     aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
     config=Config(signature_version='s3v4')
@@ -37,7 +36,6 @@ model_upload_dir = "gradio_demo_model/"
 def load_image_from_url(image_url):
     try:
         response = requests.get(image_url)
-        # Ensure the response is an image by checking the content type
         if "image" in response.headers["Content-Type"]:
             img = Image.open(BytesIO(response.content))
             return img
@@ -129,14 +127,11 @@ def display_image(image, image_url):
         response = requests.post(MODAL_INFERENCE_ENDPOINT_URL, json=payload)
         if response.status_code == 200:
             result_data = response.json()
-            for key in result_data.keys():
-                if isinstance(result_data[key], str) and result_data[key].startswith('http'):
-                    print(result_data[key])
-                    presigned = generate_presigned_url(result_data[key])
-                    response = requests.get(presigned)
-                    img = Image.open(BytesIO(response.content))
-                    img_resized = img.resize((512, 682))
-                    return img_resized
+            url = result_data["url"]
+            response = requests.get(url)
+            img = Image.open(BytesIO(response.content))
+            img_resized = img.resize((512, 682))
+            return img_resized
         else:
             results.append({"error": f"Failed to process the garment image. Status Code: {response.status_code}"})
     except requests.exceptions.RequestException as e: