Update app.py

app.py

@@ -12,7 +12,7 @@ import json
 API_URL = "https://api.openai.com/v1/chat/completions"
 
 # Set the OpenAI API Key securely (from Hugging Face secrets)
-OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+OPENAI_API_KEY = st.secrets["YOUR_OPENAI_API_KEY"]
 
 # API Call Limits (To Protect Against Abuse)
 MAX_REQUESTS_PER_SESSION = 3  # Maximum API calls per user session
@@ -133,6 +133,14 @@ initial_messages = [{
 
 # Function to encode images to base64 for API transmission
 def encode_image_to_base64(image):
+    # Convert RGBA images to RGB for JPEG compatibility
+    if image.mode == 'RGBA':
+        # Create a white background image
+        background = Image.new('RGB', image.size, (255, 255, 255))
+        # Paste the image on the background using the alpha channel
+        background.paste(image, (0, 0), image)
+        image = background
+    
     buffered = BytesIO()
     image.save(buffered, format="JPEG")
     return base64.b64encode(buffered.getvalue()).decode('utf-8')