Update app.py

app.py

@@ -5,15 +5,26 @@ from PIL import Image
 import io
 import base64
 
-# 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"
+# Configure API (Keeping DeepSeek for simplicity, but defaulting to OpenAI)
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
+
+# Check which API key is available
+if OPENAI_API_KEY:
+    API_KEY = OPENAI_API_KEY
+    API_ENDPOINT = "https://api.openai.com/v1/chat/completions"
+elif DEEPSEEK_API_KEY:
+    API_KEY = DEEPSEEK_API_KEY
+    API_ENDPOINT = "https://api.deepseek.com/v1/chat/completions"
+else:
+    API_KEY = None
+    API_ENDPOINT = None
 
 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
+    # Convert RGBA images to RGB before saving
     if image.mode != "RGB":
         image = image.convert("RGB")
     
@@ -22,7 +33,8 @@ 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:
+    
+    if not API_KEY or not API_ENDPOINT:
         st.error("❌ API key is missing! Please set it in your Hugging Face secrets.")
         return None
 
@@ -43,7 +55,7 @@ def get_valuation(item_description, image_base64=None):
     }
 
     data = {
-        "model": "gpt-4",
+        "model": "gpt-4" if "openai" in API_ENDPOINT else "deepseek-model",  # Adjust for DeepSeek
         "messages": [
             {"role": "system", "content": "You are a seasoned auction expert specializing in item valuation and auction advisement."},
             {"role": "user", "content": prompt}
@@ -88,7 +100,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 encoding
+        # Encode image for AI processing
         image_base64 = encode_image(image)
 
 if st.button("Get Estimated Value"):