Update app.py

app.py

@@ -11,6 +11,8 @@ class SynthIDApp:
     
     def login(self, hf_token):
         """Initialize the API headers with authentication."""
+        if not hf_token or not hf_token.startswith('hf_'):
+            return "Error: Please enter a valid Hugging Face token (starts with 'hf_')"
         try:
             self.headers = {"Authorization": f"Bearer {hf_token}"}
             
@@ -18,14 +20,23 @@ class SynthIDApp:
             response = requests.post(
                 self.api_url,
                 headers=self.headers,
-                json={"inputs": "Test", "parameters": {"max_new_tokens": 1}}
+                json={"inputs": "Test", "parameters": {"max_new_tokens": 1}},
+                timeout=10  # Add 10 second timeout
             )
             response.raise_for_status()
             
             return "API connection initialized successfully!"
         except Exception as e:
             self.headers = None
-            return f"Error initializing API: {str(e)}"
+            error_msg = str(e)
+            if "timeout" in error_msg.lower():
+                return "Error: API connection timed out. Please try again."
+            elif "forbidden" in error_msg.lower():
+                return "Error: Invalid token or insufficient permissions."
+            elif "not found" in error_msg.lower():
+                return "Error: Model not found or unavailable."
+            else:
+                return f"Error initializing API: {error_msg}"
 
     def update_watermark_config(self, ngram_len):
         """Update the watermarking configuration with new ngram_len."""
@@ -66,7 +77,8 @@ class SynthIDApp:
             response = requests.post(
                 self.api_url,
                 headers=self.headers,
-                json=params
+                json=params,
+                timeout=30  # Add 30 second timeout for generation
             )
             response.raise_for_status()