Update app.py

app.py

@@ -33,20 +33,11 @@ class SynthIDApp:
             
         try:
             # Prepare the API request parameters
-            # Calculate input length in tokens first
-            test_response = requests.post(
-                self.api_url,
-                headers=self.headers,
-                json={"inputs": text}
-            )
-            input_length = len(test_response.json()[0]['tokens'])
-
             # Prepare the API request parameters for watermarking
             params = {
                 "inputs": text,
                 "parameters": {
-                    "max_length": input_length,  # Limit to input length
-                    "min_length": input_length,  # Force exact length
+                    "max_new_tokens": 0,  # Don't generate new tokens
                     "do_sample": True,
                     "temperature": 0.7,
                     "top_p": 0.9,
@@ -69,24 +60,30 @@ class SynthIDApp:
             response = requests.post(
                 self.api_url,
                 headers=self.headers,
-                json=params
+                json=params,
+                timeout=30  # Add timeout
             )
             response.raise_for_status()
             
             # Extract the watermarked text
             result = response.json()
             if isinstance(result, list) and len(result) > 0:
+                if 'error' in result[0]:
+                    return text, f"API Error: {result[0]['error']}"
+                
                 watermarked_text = result[0].get('generated_text', '').strip()
                 if not watermarked_text:
                     return text, "Error: No watermarked text generated"
                 
-                # Verify we're not getting repetition
-                if len(watermarked_text.split()) > len(text.split()) * 1.5:
-                    return text, "Error: Generated text too long. Please try again."
+                # Add basic length validation
+                if len(watermarked_text) < len(text) * 0.5:
+                    return text, "Error: Generated text too short"
+                if len(watermarked_text) > len(text) * 1.5:
+                    return text, "Error: Generated text too long"
                 
                 return watermarked_text, f"Watermark applied successfully! (ngram_len: {ngram_len})"
             else:
-                return text, "Error: Unexpected API response format"
+                return text, f"Error: Unexpected API response format: {str(result)}"
                 
             return watermarked_text, f"Watermark applied successfully! (ngram_len: {ngram_len})"
         except Exception as e: