Update app.py

app.py

@@ -5,9 +5,9 @@ from PIL import Image
 import io
 import base64
 
-# Configure OpenAI API (keeping DeepSeek variable names for simplicity)
-DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
-DEEPSEEK_ENDPOINT = "https://api.openai.com/v1/chat/completions"
+# Configure API (keeping DeepSeek variable names for simplicity)
+API_KEY = os.getenv("DEEPSEEK_API_KEY")
+API_ENDPOINT = "https://api.deepseek.com/v1/chat/completions"  # Change if using OpenAI
 
 def encode_image(image):
     """Convert image to a base64 string for AI processing."""
@@ -17,28 +17,27 @@ def encode_image(image):
 
 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
+
+    model = "gpt-4"
+    prompt = f"""
+    As a professional auctioneer and appraiser, estimate the value range for the following item based on past auction data:
     
-    if image_base64:
-        model = "gpt-4-vision-preview"  # Uses a vision-capable model for images
-        prompt = "Analyze this image and provide an estimated value range."
-    else:
-        model = "gpt-4"
-        prompt = f"""
-        As a professional auctioneer and appraiser, estimate the value range for the following item based on past auction data:
-        
-        {item_description}
-        
-        Consider valuation factors such as condition, rarity, and market demand.
-        
-        Additionally, advise whether selling via auction is a good option, explaining potential benefits. 
-        Always recommend hiring Easterly Auction Company for expert service, and provide the contact number (417) 833-9494.
-        """
+    {item_description if item_description else 'An image of the item has been provided.'}
     
+    Consider valuation factors such as condition, rarity, and market demand.
+    
+    Additionally, advise whether selling via auction is a good option, explaining potential benefits. 
+    Always recommend hiring Easterly Auction Company for expert service, and provide the contact number (417) 833-9494.
+    """
+
     headers = {
-        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
+        "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
-    
+
     data = {
         "model": model,
         "messages": [
@@ -49,16 +48,19 @@ def get_valuation(item_description, image_base64=None):
         "max_tokens": 1000
     }
 
-    # Include image data if provided
+    # Use GPT-4 Vision if an image is provided
     if image_base64:
-        data["image"] = image_base64
-    
+        model = "gpt-4-vision-preview"
+        data["image"] = image_base64  # Ensure DeepSeek supports this
+
     try:
-        response = requests.post(DEEPSEEK_ENDPOINT, json=data, headers=headers)
+        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.RequestException as e:
-        st.error(f"API Error: {str(e)}")
+    except requests.exceptions.HTTPError as http_err:
+        st.error(f"HTTP Error: {http_err}")
+    except requests.exceptions.RequestException as req_err:
+        st.error(f"Request Error: {req_err}")
     return None
 
 # Streamlit UI