Update app.py

app.py

@@ -3,17 +3,24 @@ import requests
 import os
 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"
 
-def get_valuation(item_description):
+def encode_image(image):
+    """Convert image to a base64 string for AI processing."""
+    img_bytes = io.BytesIO()
+    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."""
     prompt = f"""
     As a professional auctioneer and appraiser, estimate the value range for the following item based on past auction data:
     
-    {item_description}
+    {item_description if item_description else 'An image of the item has been provided.'}
     
     Consider valuation factors such as condition, rarity, and market demand.
     
@@ -36,6 +43,10 @@ def get_valuation(item_description):
         "max_tokens": 1000
     }
     
+    # If an image is provided, include it in the request (future AI models could process it)
+    if image_base64:
+        data["image"] = image_base64
+    
     try:
         response = requests.post(DEEPSEEK_ENDPOINT, json=data, headers=headers)
         response.raise_for_status()
@@ -53,6 +64,8 @@ st.markdown("### Upload an image or describe an item to estimate its value")
 input_method = st.radio("Choose input method:", ["Describe the item", "Upload an image"])
 
 description = ""
+image_base64 = None
+
 if input_method == "Describe the item":
     description = st.text_area("Describe the item in detail*")
 
@@ -67,17 +80,14 @@ elif input_method == "Upload an image":
         if image.mode == "RGBA":
             image = image.convert("RGB")
         
-        img_bytes = io.BytesIO()
-        image.save(img_bytes, format="JPEG")
-        img_bytes = img_bytes.getvalue()
-        # Image processing could be added here in the future
+        # Encode image for AI processing
+        image_base64 = encode_image(image)
 
 if st.button("Get Estimated Value"):
-    if not description:
+    if not description and not image_base64:
         st.error("Please provide a description or upload an image.")
     else:
         with st.spinner("Analyzing item and fetching valuation..."):
-            valuation = get_valuation(description)
+            valuation = get_valuation(description, image_base64)
             if valuation:
                 st.markdown(valuation)
-