Update app.py

app.py

@@ -36,10 +36,11 @@ class SynthIDApp:
             params = {
                 "inputs": text,
                 "parameters": {
-                    "max_new_tokens": 100,
+                    "return_full_text": True,
                     "do_sample": True,
                     "temperature": 0.7,
                     "top_p": 0.9,
+                    "max_length": None,  # Use input length
                     "watermarking_config": {
                         "keys": self.WATERMARK_KEYS,
                         "ngram_len": int(ngram_len)  # Ensure integer
@@ -55,12 +56,16 @@ class SynthIDApp:
             )
             response.raise_for_status()
             
-            # Extract the generated text
+            # Extract the watermarked text
             result = response.json()
             if isinstance(result, list) and len(result) > 0:
-                watermarked_text = result[0].get('generated_text', text)
+                watermarked_text = result[0].get('generated_text', '')
+                if not watermarked_text:
+                    return text, "Error: No watermarked text generated"
+                # Clean up any extra whitespace
+                watermarked_text = watermarked_text.strip()
             else:
-                watermarked_text = text
+                return text, "Error: Unexpected API response format"
                 
             return watermarked_text, f"Watermark applied successfully! (ngram_len: {ngram_len})"
         except Exception as e:
@@ -108,7 +113,8 @@ with gr.Blocks(title="SynthID Text Watermarking Tool") as app:
                 input_text = gr.Textbox(
                     label="Input Text", 
                     lines=5,
-                    placeholder="Enter text to watermark..."
+                    placeholder="Enter text to watermark...",
+                    value="The quick brown fox jumps over the lazy dog."
                 )
                 output_text = gr.Textbox(label="Watermarked Text", lines=5)
             with gr.Column(scale=1):