Update app.py

app.py

@@ -5,9 +5,9 @@ from PIL import Image
 import io
 import base64
 
-# 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
+# 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
 
 def encode_image(image):
     """Convert image to a base64 string for AI processing."""
@@ -17,11 +17,11 @@ 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.")
+        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:
     
@@ -39,7 +39,7 @@ def get_valuation(item_description, image_base64=None):
     }
 
     data = {
-        "model": model,
+        "model": "gpt-4",
         "messages": [
             {"role": "system", "content": "You are a seasoned auction expert specializing in item valuation and auction advisement."},
             {"role": "user", "content": prompt}
@@ -48,10 +48,9 @@ def get_valuation(item_description, image_base64=None):
         "max_tokens": 1000
     }
 
-    # Use GPT-4 Vision if an image is provided
+    # Ensure DeepSeek supports image inputs before using it
     if image_base64:
-        model = "gpt-4-vision-preview"
-        data["image"] = image_base64  # Ensure DeepSeek supports this
+        data["image"] = image_base64
 
     try:
         response = requests.post(API_ENDPOINT, json=data, headers=headers)
@@ -83,7 +82,7 @@ elif input_method == "Upload an image":
         image = Image.open(uploaded_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")