Update app.py

app.py

@@ -3,16 +3,20 @@ from huggingface_hub import InferenceClient, HfApi
 from huggingface_hub.utils import HfHubHTTPError
 import os
 
-def check_and_enable_api(model_id):
+def check_api_status(model_id):
     try:
-        api = HfApi()
-        model_info = api.model_info(model_id)
-        if not model_info.api:
+        client = InferenceClient(model_id)
+        # Simple test request to check if API is responsive
+        client.health_check()
+        return "API is accessible"
+    except Exception as e:
+        try:
+            # Try to enable the API
+            api = HfApi()
             api.toggle_model_api(model_id, True)
             return "API was disabled. Now enabled for the model."
-        return "API already enabled for the model."
-    except Exception as e:
-        return f"Error checking API status: {str(e)}"
+        except Exception as enable_err:
+            return f"Could not enable API: {str(enable_err)}"
 
 def get_api_status():
     token = os.getenv('HF_TOKEN')
@@ -22,11 +26,9 @@ def get_api_status():
         return "⚠️ No API token found. Please set HF_TOKEN environment variable."
     
     try:
-        status = check_and_enable_api(model_id)
         client = InferenceClient(model_id, token=token)
+        status = check_api_status(model_id)
         return f"✅ Connected to {model_id} | {status}"
-    except HfHubHTTPError as e:
-        return f"❌ API Error: {str(e)}"
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
@@ -105,8 +107,8 @@ with gr.Blocks() as demo:
         ],
     )
     
-    # Add API status at the footer
-    gr.HTML(
+    # Add API status at the footer with error handling
+    footer = gr.HTML(
         value=f"<div style='text-align: center; padding: 10px; background-color: #f0f0f0; border-top: 1px solid #ddd;'>{get_api_status()}</div>",
         every=30  # Updates every 30 seconds
     )