Update app.py

app.py

@@ -86,22 +86,49 @@ def duckduckgo_search_api(query):
 
 @st.cache_data(show_spinner=False, ttl=300)
 def search_image(query):
-    """MASTER FUNCTION: Google Search -> DuckDuckGo"""
-    cx = os.environ.get("GOOGLE_CX", st.secrets.get("GOOGLE_CX", ""))
-    
-    key1 = os.environ.get("GOOGLE_SEARCH_KEY", st.secrets.get("GOOGLE_SEARCH_KEY", ""))
-    if key1 and cx:
-        url = google_search_api(query, key1, cx)
-        if url: 
-            return url
+    """MASTER FUNCTION: Google Search -> DuckDuckGo - FIXED FOR HUGGING FACE"""
+    try:
+        # Try to get secrets from Hugging Face Spaces
+        # Method 1: Check st.secrets (Hugging Face Spaces method)
+        try:
+            if hasattr(st, 'secrets'):
+                # Hugging Face Spaces stores secrets in st.secrets
+                cx = st.secrets.get("GOOGLE_CX", "")
+                key1 = st.secrets.get("GOOGLE_SEARCH_KEY", "")
+                key2 = st.secrets.get("GOOGLE_SEARCH_KEY_2", "")
+                
+                if key1 and cx:
+                    url = google_search_api(query, key1, cx)
+                    if url: 
+                        return url
+                
+                if key2 and cx:
+                    url = google_search_api(query, key2, cx)
+                    if url: 
+                        return url
+        except:
+            pass
+        
+        # Method 2: Check environment variables
+        cx = os.environ.get("GOOGLE_CX", "")
+        key1 = os.environ.get("GOOGLE_SEARCH_KEY", "")
+        key2 = os.environ.get("GOOGLE_SEARCH_KEY_2", "")
+        
+        if key1 and cx:
+            url = google_search_api(query, key1, cx)
+            if url: 
+                return url
 
-    key2 = os.environ.get("GOOGLE_SEARCH_KEY_2", st.secrets.get("GOOGLE_SEARCH_KEY_2", ""))
-    if key2 and cx:
-        url = google_search_api(query, key2, cx)
-        if url: 
-            return url
+        if key2 and cx:
+            url = google_search_api(query, key2, cx)
+            if url: 
+                return url
 
-    return duckduckgo_search_api(query)
+        # Method 3: Fallback to DuckDuckGo
+        return duckduckgo_search_api(query)
+        
+    except Exception as e:
+        return f"Search Error: {str(e)[:100]}"
 
 # Graph plotting function - REORDERED AND FIXED
 def execute_plotting_code(code_snippet):
@@ -457,7 +484,37 @@ with st.sidebar:
         "**Free Tier:** 10,000 requests/month\n"
         "**Models:** Llama 3.1 8B Instant (fast & free)"
     )
+
+    # Image Search Status
+    st.divider()
+    st.header("🖼️ Image Search")
     
+    # Check if Google Search is configured
+    google_configured = False
+    try:
+        if hasattr(st, 'secrets'):
+            if st.secrets.get("GOOGLE_SEARCH_KEY") and st.secrets.get("GOOGLE_CX"):
+                google_configured = True
+        elif os.environ.get("GOOGLE_SEARCH_KEY") and os.environ.get("GOOGLE_CX"):
+            google_configured = True
+    except:
+        pass
+    
+    if google_configured:
+        st.success("✅ Google Image Search configured")
+        st.caption("Will use Google Custom Search API")
+    else:
+        st.warning("⚠️ Google Image Search not configured")
+        st.caption("Will use DuckDuckGo (free but limited)")
+        st.info("""
+To enable Google Image Search:
+1. Go to [Google Cloud Console](https://console.cloud.google.com)
+2. Enable Custom Search API
+3. Get API Key and Search Engine ID (CX)
+4. Add to Hugging Face Secrets:
+   - GOOGLE_SEARCH_KEY = "your-api-key"
+   - GOOGLE_CX = "your-search-engine-id"
+""")
     # Quick test button
     if api_key and st.button("🔍 Test API Key", use_container_width=True):
         with st.spinner("Testing connection..."):