Update app.py

app.py

@@ -43,15 +43,19 @@ def create_building_analysis(input_image):
     except Exception as e:
         return create_info_image(f"Analysis error: {str(e)}")
 
+
+# ✅ FIXED FUNCTION
 def test_huggingface_api():
-    """Test if Hugging Face API is working"""
+    """Test if Hugging Face API is working correctly"""
     if not HF_TOKEN:
         return False, "No HF_TOKEN found in environment variables"
     
-    # Test with a simple model
+    # Use an image model for testing
     API_URL = "https://api-inference.huggingface.co/models/google/vit-base-patch16-224"
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    payload = {"inputs": "Hello"}
+    
+    # ✅ Use a valid image URL instead of a text string
+    payload = {"inputs": "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"}
     
     try:
         response = requests.post(API_URL, headers=headers, json=payload, timeout=10)
@@ -62,6 +66,7 @@ def test_huggingface_api():
     except Exception as e:
         return False, f"❌ API test error: {str(e)}"
 
+
 def generate_architectural_remix(original_image, style_choice):
     """Main pipeline"""
     if original_image is None:
@@ -101,6 +106,7 @@ def generate_architectural_remix(original_image, style_choice):
         status = f"❌ Pipeline error: {str(e)}"
         return building_analysis, status, error_img
 
+
 def generate_with_api(style_name):
     """Try to generate image via API"""
     if not HF_TOKEN:
@@ -141,6 +147,7 @@ def generate_with_api(style_name):
         print(f"❌ Request error: {str(e)}")
         return None
 
+
 def create_demo_building(style_name):
     """Create a beautiful demo building"""
     # Style-based color schemes
@@ -210,6 +217,7 @@ def create_demo_building(style_name):
     
     return img
 
+
 def create_info_image(message):
     """Create informational image"""
     img = Image.new('RGB', (512, 512), color='#2C3E50')
@@ -239,6 +247,7 @@ def create_info_image(message):
     
     return img
 
+
 # Create Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
@@ -323,4 +332,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     """)
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)