Update app.py

app.py

@@ -5,19 +5,23 @@ from PIL import Image
 import io
 import base64
 
-# Configure API (use OpenAI or DeepSeek)
-API_KEY = os.getenv("DEEPSEEK_API_KEY")  # Change to "OPENAI_API_KEY" if using OpenAI
-API_ENDPOINT = "https://api.openai.com/v1/chat/completions"  # Adjust if using DeepSeek
+# Configure OpenAI API (keeping DeepSeek variable names for simplicity)
+API_KEY = os.getenv("DEEPSEEK_API_KEY")  # Ensure this is set in Hugging Face secrets
+API_ENDPOINT = "https://api.openai.com/v1/chat/completions"
 
 def encode_image(image):
     """Convert image to a base64 string for AI processing."""
     img_bytes = io.BytesIO()
+    
+    # Convert to RGB to avoid RGBA-related errors
+    if image.mode != "RGB":
+        image = image.convert("RGB")
+    
     image.save(img_bytes, format="JPEG")
     return base64.b64encode(img_bytes.getvalue()).decode()
 
 def get_valuation(item_description, image_base64=None):
     """Generate an estimated value range based on past auction data."""
-    
     if not API_KEY:
         st.error("❌ API key is missing! Please set it in your Hugging Face secrets.")
         return None
@@ -48,16 +52,17 @@ def get_valuation(item_description, image_base64=None):
         "max_tokens": 1000
     }
 
-    # Ensure DeepSeek supports image inputs before using it
-    if image_base64:
-        data["image"] = image_base64
-
     try:
         response = requests.post(API_ENDPOINT, json=data, headers=headers)
         response.raise_for_status()
         return response.json().get("choices", [{}])[0].get("message", {}).get("content", "No response from API.")
     except requests.exceptions.HTTPError as http_err:
-        st.error(f"HTTP Error: {http_err}")
+        if response.status_code == 401:
+            st.error("❌ Unauthorized: Check your API key in Hugging Face secrets.")
+        elif response.status_code == 404:
+            st.error("❌ API Endpoint Not Found: Verify the API URL.")
+        else:
+            st.error(f"HTTP Error: {http_err}")
     except requests.exceptions.RequestException as req_err:
         st.error(f"Request Error: {req_err}")
     return None
@@ -83,11 +88,7 @@ elif input_method == "Upload an image":
         st.image(image, caption="Uploaded Image", use_column_width=True)
         description = st.text_area("(Optional) Add additional details about the item")
 
-        # Convert image to RGB mode before saving as JPEG
-        if image.mode == "RGBA":
-            image = image.convert("RGB")
-        
-        # Encode image for AI processing
+        # Convert image to RGB mode before encoding
         image_base64 = encode_image(image)
 
 if st.button("Get Estimated Value"):