Update app.py

app.py

@@ -28,20 +28,20 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-# --- Configuration ---
+# --- API Configuration ---
 API_URL = "https://api.openai.com/v1/chat/completions"
 
-# --- Cache utilities ---
+# --- Caching helpers ---
 @st.cache_data
 def encode_image_to_b64(image: Image.Image) -> str:
     # Convert to RGB if needed
     if image.mode == 'RGBA':
-        bg = Image.new('RGB', image.size, (255,255,255))
+        bg = Image.new('RGB', image.size, (255, 255, 255))
         bg.paste(image, mask=image.split()[3])
         image = bg
     elif image.mode not in ['RGB', 'L']:
         image = image.convert('RGB')
-    # Resize to limit token usage
+    # Resize large images
     max_size = (1024, 1024)
     if image.width > max_size[0] or image.height > max_size[1]:
         image.thumbnail(max_size, Image.LANCZOS)
@@ -49,23 +49,27 @@ def encode_image_to_b64(image: Image.Image) -> str:
     image.save(buf, format='JPEG', quality=85)
     return base64.b64encode(buf.getvalue()).decode('utf-8')
 
+
 def fix_formatting(text: str) -> str:
     text = re.sub(r'(\d+)([A-Za-z])', r'\1 \2', text)
     text = re.sub(r'([.,])([A-Za-z0-9])', r'\1 \2', text)
     text = re.sub(r':([A-Za-z0-9])', r': \1', text)
     return text
 
-# --- Analysis function via requests ---
+# --- Core analysis ---
 def analyze_home_photos(images, timeframe, details, api_key):
     if not api_key:
         return "Error: API key is required."
-    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
-    # Build system prompt
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
+    }
+    # System prompt
     system_prompt = (
         f"You are a real estate advisor. Today's date is {datetime.now().strftime('%B %d, %Y')}. "
-        "Analyze only visible areas in the images and give hyper-specific recommendations."
+        "Analyze only visible areas and give hyper-specific recommendations."
     )
-    # Build user content with embedded images
+    # User content with embedded images
     user_content = f"Sell within {timeframe}. {details}\n\n"
     for img in images:
         b64 = encode_image_to_b64(img)
@@ -81,29 +85,34 @@ def analyze_home_photos(images, timeframe, details, api_key):
     resp = requests.post(API_URL, headers=headers, json=payload, timeout=90)
     if resp.status_code == 200:
         data = resp.json()
-        text = data['choices'][0]['message']['content']
-        return fix_formatting(text)
+        return fix_formatting(data['choices'][0]['message']['content'])
     return f"API Error {resp.status_code}: {resp.text[:200]}"
 
 # --- Cost calculation ---
 def calculate_cost(usage):
     if not usage:
         return 0.0
-    in_tok = usage.get('prompt_tokens',0)
-    out_tok = usage.get('completion_tokens',0)
-    cost = in_tok/1e6*3 + out_tok/1e6*10
-    return cost
+    in_tok = usage.get('prompt_tokens', 0)
+    out_tok = usage.get('completion_tokens', 0)
+    return in_tok/1e6*3 + out_tok/1e6*10
 
-# --- App layout ---
+# --- UI layout ---
 st.title("Home Value Maximizer 🏡")
 
 # Sidebar inputs
 api_key = st.sidebar.text_input("OpenAI API Key", type="password")
-uploaded = st.sidebar.file_uploader("Upload home photos", accept_multiple_files=True, type=["jpg","png","jpeg"])
-
-timeframe = st.sidebar.selectbox("Selling timeframe", ["Within 1 month","1–3 months","3–6 months","6–12 months",">12 months"])
+uploaded = st.sidebar.file_uploader(
+    "Upload home photos", accept_multiple_files=True, type=["jpg", "jpeg", "png"]
+)
+timeframe = st.sidebar.selectbox(
+    "Selling timeframe", ["Within 1 month", "1–3 months", "3–6 months", "6–12 months", ">12 months"]
+)
 budget = st.sidebar.slider("Max improvement budget ($)", 1000, 50000, 10000, 1000)
-focus = st.sidebar.multiselect("Focus areas", ["Curb appeal","Kitchen","Bathroom","Living Spaces","Outdoor","Storage"], default=["Kitchen","Bathroom"])
+focus = st.sidebar.multiselect(
+    "Focus areas",
+    ["Curb appeal", "Kitchen", "Bathroom", "Living Spaces", "Outdoor", "Storage"],
+    default=["Kitchen", "Bathroom"]
+)
 
 if st.sidebar.button("🔍 Analyze My Home"):
     if not api_key:
@@ -113,14 +122,23 @@ if st.sidebar.button("🔍 Analyze My Home"):
     else:
         images = []
         for f in uploaded:
-            try: images.append(Image.open(f))
-            except: pass
+            try:
+                images.append(Image.open(f))
+            except:
+                pass
         with st.spinner("Analyzing..."):
-            result = analyze_home_photos(images, timeframe, f"Budget ${budget}. Focus: {', '.join(focus)}", api_key)
-        st.markdown(f"<div style='background:#e3f2fd;padding:1rem;border-left:4px solid #1f77b4;'>{result}</div>", unsafe_allow_html=True)
-        # Cost display
-        # Note: HF Spaces does not return usage, so this may be blank
-        usage = {}  # placeholder
+            result = analyze_home_photos(
+                images,
+                timeframe,
+                f"Budget ${budget}. Focus: {', '.join(focus)}",
+                api_key
+            )
+        # Render result with styled container
+        st.markdown(
+            f"<div style='background:#e3f2fd;padding:1rem;border-left:4px solid #1f77b4;'>{result}</div>",
+            unsafe_allow_html=True
+        )
+        # Display cost (usage info not returned in HF)
+        usage = {}
         cost = calculate_cost(usage)
         st.metric("💲 Estimated cost", f"${cost:.2f}")
-```