Update app.py

app.py

@@ -12,7 +12,15 @@ import json
 API_URL = "https://api.openai.com/v1/chat/completions"
 
 # Set the OpenAI API Key securely (from Hugging Face secrets)
-OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+try:
+    OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+except:
+    # For testing, can use this but NEVER commit actual keys
+    OPENAI_API_KEY = "YOUR_API_KEY_HERE"
+    st.warning("⚠️ Using demo mode - API calls will not work. Add your API key to secrets.")
+    
+# Define which model to use - use a model that supports vision capabilities
+OPENAI_MODEL = "gpt-4-vision-preview"  # The correct model name for vision capabilities
 
 # API Call Limits (To Protect Against Abuse)
 MAX_REQUESTS_PER_SESSION = 3  # Maximum API calls per user session
@@ -165,26 +173,46 @@ def call_openai_api(messages):
     }
     
     payload = {
-        "model": "gpt-4-vision-preview",
+        "model": OPENAI_MODEL,
         "messages": messages,
         "max_tokens": 1500
     }
     
     try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()  # Raise exception for 4XX/5XX status codes
-        
-        response_data = response.json()
-        if response_data and "choices" in response_data:
-            st.session_state["api_calls"] += 1
-            st.session_state["last_request_time"] = time.time()
-            return response_data["choices"][0]["message"]["content"]
-        else:
-            st.error("❌ API response format unexpected. Please try again.")
-            return None
+        with st.status("Analyzing your home photos..."):
+            st.write("Connecting to AI service...")
+            response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
+            
+            # Check for common error cases
+            if response.status_code == 404:
+                st.error("❌ API endpoint not found. The model may not be available or the URL is incorrect.")
+                return None
+            elif response.status_code == 401:
+                st.error("❌ Authentication failed. Please check your API key.")
+                return None
+            elif response.status_code == 429:
+                st.error("❌ Rate limit exceeded. Please try again later.")
+                return None
+            
+            response.raise_for_status()  # Raise exception for other 4XX/5XX status codes
+            
+            st.write("Processing analysis results...")
+            response_data = response.json()
+            
+            if response_data and "choices" in response_data:
+                st.session_state["api_calls"] += 1
+                st.session_state["last_request_time"] = time.time()
+                return response_data["choices"][0]["message"]["content"]
+            else:
+                st.error("❌ API response format unexpected. Please try again.")
+                st.write(f"Response: {response_data}")
+                return None
     except requests.exceptions.RequestException as e:
         st.error(f"❌ API request failed: {str(e)}")
         return None
+    except Exception as e:
+        st.error(f"❌ An unexpected error occurred: {str(e)}")
+        return None
 
 # Function to generate home improvement recommendations
 def generate_recommendations(images, property_type, price_range, timeframe, additional_details, messages):