Update app.py

app.py

@@ -47,6 +47,61 @@ if "debug" not in st.session_state:
 with st.sidebar:
     st.session_state.debug = st.toggle("Enable Debug Mode", value=debug_mode)
 
+# Add after existing debug mode toggle in sidebar
+with st.sidebar:
+    if st.session_state.debug:
+        st.write("### Connection Diagnostics")
+        if st.button("Test File Upload Connection"):
+            try:
+                # Create a simple test file in memory
+                import io
+                test_file = io.BytesIO(b"test content")
+                test_file.name = "test.txt"
+                
+                # Test the Streamlit file uploader connection
+                st.write("Checking file upload capability...")
+                st.write("Status: Testing... If this freezes, there may be connectivity issues.")
+                
+                # Check basic file operations
+                test_path = "test_upload_capability.txt"
+                try:
+                    with open(test_path, "w") as f:
+                        f.write("test")
+                    st.write("✅ File write test: Success")
+                    import os
+                    os.remove(test_path)
+                    st.write("✅ File delete test: Success")
+                except Exception as e:
+                    st.write(f"❌ File operation test: Failed - {str(e)}")
+                
+                # Check Streamlit session state
+                try:
+                    st.session_state.test_value = "test"
+                    if st.session_state.test_value == "test":
+                        st.write("✅ Session state test: Success")
+                except Exception as e:
+                    st.write(f"❌ Session state test: Failed - {str(e)}")
+                
+                # Environment variables check
+                import os
+                st.write("### Environment Variables")
+                for key in ["STREAMLIT_SERVER_ENABLE_CORS", "STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION", 
+                            "TEMP", "TMP", "TMPDIR"]:
+                    st.write(f"{key}: {os.environ.get(key, 'Not set')}")
+                
+                # Check for specific Hugging Face Spaces environment variables
+                hf_vars = [k for k in os.environ if k.startswith("HF_")]
+                if hf_vars:
+                    st.write("### Hugging Face Environment Variables")
+                    for key in hf_vars:
+                        st.write(f"{key}: {os.environ.get(key, 'Not set')}")
+                
+                st.success("Diagnostics completed!")
+            except Exception as e:
+                st.error(f"Diagnostics error: {str(e)}")
+                import traceback
+                st.error(traceback.format_exc())
+
 def log_debug(message):
     """Helper function to log debug messages only when debug mode is enabled"""
     if st.session_state.debug:
@@ -706,21 +761,82 @@ def main():
             if url and url.strip():
                 try:
                     import requests
-                    # Simplified URL handling
+                    # Simplified URL handling with more robust approach
                     headers = {
-                        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+                        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+                        'Accept': 'image/jpeg, image/png, image/*, */*',
+                        'Referer': 'https://huggingface.co/'
                     }
-                    response = requests.get(url, stream=True, headers=headers, timeout=10)
                     
-                    if response.status_code == 200:
-                        # Process directly as in original code
-                        image = Image.open(io.BytesIO(response.content)).convert("RGB")
-                        uploaded_image = image
-                        st.session_state.upload_method = "url"
-                    else:
-                        st.error(f"Failed to load image from URL: Status code {response.status_code}")
+                    # Try three different methods to handle various API restrictions
+                    try_methods = True
+                    
+                    # Method 1: Direct requests
+                    if try_methods:
+                        try:
+                            response = requests.get(url, stream=True, headers=headers, timeout=10)
+                            if response.status_code == 200 and 'image' in response.headers.get('Content-Type', ''):
+                                try:
+                                    image = Image.open(io.BytesIO(response.content)).convert("RGB")
+                                    uploaded_image = image
+                                    st.session_state.upload_method = "url_direct"
+                                    try_methods = False
+                                    st.success("✅ Image loaded via direct request")
+                                except Exception as e:
+                                    st.warning(f"Direct method received data but couldn't process as image: {str(e)}")
+                            else:
+                                st.info(f"Direct method failed: Status {response.status_code}, trying alternative method...")
+                        except Exception as e:
+                            st.info(f"Direct method error: {str(e)}, trying alternative method...")
+                    
+                    # Method 2: Use Python's urllib as fallback
+                    if try_methods:
+                        try:
+                            import urllib.request
+                            from urllib.error import HTTPError
+                            
+                            opener = urllib.request.build_opener()
+                            opener.addheaders = [('User-agent', headers['User-Agent'])]
+                            urllib.request.install_opener(opener)
+                            
+                            with urllib.request.urlopen(url, timeout=10) as response:
+                                image_data = response.read()
+                                image = Image.open(io.BytesIO(image_data)).convert("RGB")
+                                uploaded_image = image
+                                st.session_state.upload_method = "url_urllib"
+                                try_methods = False
+                                st.success("✅ Image loaded via urllib")
+                        except HTTPError as e:
+                            st.info(f"urllib method failed: HTTP error {e.code}, trying next method...")
+                        except Exception as e:
+                            st.info(f"urllib method error: {str(e)}, trying next method...")
+                    
+                    # Method 3: Use a proxy service as last resort
+                    if try_methods:
+                        try:
+                            # This uses an image proxy service to bypass CORS issues
+                            # Only as last resort since it depends on external service
+                            proxy_url = f"https://images.weserv.nl/?url={url}"
+                            response = requests.get(proxy_url, stream=True, timeout=10)
+                            if response.status_code == 200:
+                                image = Image.open(io.BytesIO(response.content)).convert("RGB")
+                                uploaded_image = image
+                                st.session_state.upload_method = "url_proxy"
+                                try_methods = False
+                                st.success("✅ Image loaded via proxy service")
+                            else:
+                                st.error(f"All methods failed to load the image from URL. Last status: {response.status_code}")
+                        except Exception as e:
+                            st.error(f"All methods failed. Final error: {str(e)}")
+                    
+                    if not uploaded_image:
+                        st.error("Failed to load image using all available methods.")
+                    
                 except Exception as e:
-                    st.error(f"Error loading image from URL: {str(e)}")
+                    st.error(f"Error processing URL: {str(e)}")
+                    if st.session_state.debug:
+                        import traceback
+                        st.error(traceback.format_exc())
         
         # If we have an uploaded image, process it
         if uploaded_image is not None: