Update app.py

app.py

@@ -1,18 +1,15 @@
 import gradio as gr
 import requests
-from transformers import SynthIDTextWatermarkingConfig
+import json
 
 class SynthIDApp:
     def __init__(self):
         self.api_url = "https://api-inference.huggingface.co/models/google/gemma-2b"
         self.headers = None
-        self.watermarking_config = None
         self.WATERMARK_KEYS = [654, 400, 836, 123, 340, 443, 597, 160, 57, 789]
     
     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}"}
             
@@ -20,34 +17,14 @@ class SynthIDApp:
             response = requests.post(
                 self.api_url,
                 headers=self.headers,
-                json={"inputs": "Test", "parameters": {"max_new_tokens": 1}},
-                timeout=10  # Add 10 second timeout
+                json={"inputs": "Test", "parameters": {"max_new_tokens": 1}}
             )
             response.raise_for_status()
             
             return "API connection initialized successfully!"
         except Exception as e:
             self.headers = None
-            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."""
-        try:
-            self.watermarking_config = SynthIDTextWatermarkingConfig(
-                keys=self.WATERMARK_KEYS,
-                ngram_len=ngram_len
-            )
-            return f"Watermark config updated: ngram_len = {ngram_len}"
-        except Exception as e:
-            return f"Error updating config: {str(e)}"
+            return f"Error initializing API: {str(e)}"
 
     def apply_watermark(self, text, ngram_len):
         """Apply SynthID watermark to input text using the inference API."""
@@ -55,9 +32,6 @@ class SynthIDApp:
             return text, "Error: API not initialized. Please login first."
             
         try:
-            # Update watermark config with current ngram_len
-            self.update_watermark_config(ngram_len)
-            
             # Prepare the API request parameters
             params = {
                 "inputs": text,
@@ -67,8 +41,8 @@ class SynthIDApp:
                     "temperature": 0.7,
                     "top_p": 0.9,
                     "watermarking_config": {
-                        "keys": self.watermarking_config.keys,
-                        "ngram_len": self.watermarking_config.ngram_len
+                        "keys": self.WATERMARK_KEYS,
+                        "ngram_len": int(ngram_len)  # Ensure integer
                     }
                 }
             }
@@ -77,8 +51,7 @@ class SynthIDApp:
             response = requests.post(
                 self.api_url,
                 headers=self.headers,
-                json=params,
-                timeout=30  # Add 30 second timeout for generation
+                json=params
             )
             response.raise_for_status()