Update app.py

app.py

@@ -8,7 +8,19 @@ import time
 import random
 import os
 
-# Function to handle the try-on process
+# Load environment variables
+tryon_url = os.environ.get('tryon_url', 'http://default-url/')  # Default URL for testing
+token = os.environ.get('token', 'default-token')
+cookie = os.environ.get('Cookie', 'default-cookie')
+referer = os.environ.get('referer', 'default-referer')
+
+headers = {
+    'Content-Type': 'application/json',
+    'token': token,
+    'Cookie': cookie,
+    'referer': referer
+}
+
 def tryon(person_img, garment_img, seed, randomize_seed):
     if person_img is None or garment_img is None:
         st.warning("Empty image")
@@ -21,11 +33,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
     encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
 
-    url = "http://" + os.environ['tryon_url'] + "Submit"
-    token = os.environ['token']
-    cookie = os.environ['Cookie']
-    referer = os.environ['referer']
-    headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
+    url = tryon_url + "Submit"
     data = {
         "clothImage": encoded_garment_img,
         "humanImage": encoded_person_img,
@@ -48,7 +56,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     info = ""
     for i in range(Max_Retry):
         try:
-            url = "http://" + os.environ['tryon_url'] + "Query?taskId=" + uuid
+            url = tryon_url + "Query?taskId=" + uuid
             response = requests.get(url, headers=headers, timeout=20)
             if response.status_code == 200:
                 result = response.json()['result']
@@ -90,7 +98,6 @@ st.markdown("""
 This project leverages the advanced virtual try-on model by Kwai-Kolors. We've created a simple and interactive GUI for users to experience virtual try-ons. For more innovative features and models, check out Kwai-Kolors on their official website!
 
 [Explore Kwai-Kolors' website](https://klingai.com/)
-
 """)
 
 # Display the demo image