Update app.py

app.py

@@ -14,17 +14,14 @@ def process_image(input_image, prompt):
     if input_image is None:
         return None, "Please upload an image"
     
-    # Convert PIL Image to bytes
-    img_byte_arr = io.BytesIO()
-    input_image.save(img_byte_arr, format='WEBP')
-    img_byte_arr = img_byte_arr.getvalue()
-    
-    # Encode to base64
-    base64_image = f"data:image/webp;base64,{base64.b64encode(img_byte_arr).decode('utf-8')}"
+    # Convert PIL Image directly to base64 string
+    buffer = io.BytesIO()
+    input_image.save(buffer, format="WEBP")
+    base64_image = f"data:image/webp;base64,{base64.b64encode(buffer.getvalue()).decode('utf-8')}"
     
     # Prepare headers and data
     headers = {
-        "Authorization": f"Bearer {os.getenv('HF_API_KEY')}",
+        "Authorization": f"Bearer {API_TOKEN}",
         "Content-Type": "application/json",
         "Accept": "application/json"
     }
@@ -45,7 +42,7 @@ def process_image(input_image, prompt):
             response = requests.post(API_URL, headers=headers, json=data, timeout=300)
             
             if response.status_code == 200:
-                # Convert the response base64 image back to PIL Image
+                # Convert the response base64 directly to PIL Image
                 result = response.json()
                 image_bytes = base64.b64decode(result["final_image"])
                 output_image = Image.open(io.BytesIO(image_bytes))
@@ -108,5 +105,5 @@ with gr.Blocks(title="Room Design Diffusion", theme=gr.themes.Soft()) as demo:
     - Be patient as generation can take a few minutes
     """)
 
-# Launch the app
+# Launch the app with sharing enabled
 demo.launch(share=True)