Update app.py

app.py

@@ -20,22 +20,19 @@ def encode_image(image):
     image.save(img_bytes, format="JPEG")
     return base64.b64encode(img_bytes.getvalue()).decode()
 
-def get_valuation(item_description, image_base64=None):
+def get_valuation(item_description, image=None):
     """Generate an estimated value range based on past auction data."""
     if not OPENAI_API_KEY:
         st.error("❌ API key is missing! Please set it in your Hugging Face secrets.")
         return None
 
-    prompt = f"""
-    As a professional auctioneer and appraiser, estimate the value range for the following item based on past auction data:
+    messages = [
+        {"role": "system", "content": "You are a seasoned auction expert specializing in item valuation and auction advisement."},
+        {"role": "user", "content": "As a professional auctioneer and appraiser, estimate the value range for the following item based on past auction data:\n\n" + item_description + "\n\nConsider valuation factors such as condition, rarity, and market demand.\n\nAdditionally, 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.
-    """
+    if image:
+        messages.append({"role": "user", "content": {"type": "image_url", "image_url": image}})
     
     headers = {
         "Authorization": f"Bearer {OPENAI_API_KEY}",
@@ -43,11 +40,8 @@ def get_valuation(item_description, image_base64=None):
     }
     
     data = {
-        "model": "gpt-4",
-        "messages": [
-            {"role": "system", "content": "You are a seasoned auction expert specializing in item valuation and auction advisement."},
-            {"role": "user", "content": prompt}
-        ],
+        "model": "gpt-4-turbo",
+        "messages": messages,
         "temperature": 0.7,
         "max_tokens": 1000
     }
@@ -76,7 +70,7 @@ 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
+image_url = None
 
 if input_method == "Describe the item":
     description = st.text_area("Describe the item in detail*")
@@ -88,14 +82,16 @@ 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")
         
-        # Encode image for AI processing
-        image_base64 = encode_image(image)
+        # Save image to a temporary file and generate a URL
+        image_path = f"temp_{uploaded_image.name}"
+        image.save(image_path)
+        image_url = f"file://{image_path}"  # Simulating an image URL for OpenAI Vision model
 
 if st.button("Get Estimated Value"):
-    if not description and not image_base64:
+    if not description and not image_url:
         st.error("Please provide a description or upload an image.")
     else:
         with st.spinner("Analyzing item and fetching valuation..."):
-            valuation = get_valuation(description, image_base64)
+            valuation = get_valuation(description, image_url)
             if valuation:
                 st.markdown(valuation)