Update app.py

app.py

@@ -44,6 +44,15 @@ def get_session_id():
 # Simplified trial management for HF Spaces
 trial_tracker = {}
 
+# Trial limit configuration (can be changed easily)
+TRIAL_LIMIT = 5  # Increased from 2 to 5 for better demo experience
+
+def clear_trial_cache():
+    """Clear all trial tracking data - useful for demos and testing."""
+    global trial_tracker
+    trial_tracker.clear()
+    print("✅ Trial cache cleared - all users can start fresh!")
+
 def get_trial_count() -> int:
     """Get current trial verification count."""
     session_id = get_session_id()
@@ -57,6 +66,9 @@ def increment_trial_count() -> int:
     trial_tracker[session_id] = new_count
     return new_count
 
+# Clear cache on startup to ensure fresh start
+clear_trial_cache()
+
 # DeBERTa API Functions
 async def call_deberta_api_async(evidence: str, claim: str, session_timeout: int = 30) -> dict:
     """Async call to DeBERTa FEVER API for fact verification."""
@@ -361,14 +373,14 @@ def verify_news_simple(input_text: str, progress=gr.Progress()) -> str:
     try:
         # Check trial limits for free users
         current_count = get_trial_count()
-        if current_count >= 2:
-            return """
+        if current_count >= TRIAL_LIMIT:
+            return f"""
             <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
                         padding: 2rem; border-radius: 12px; color: white; text-align: center;
                         box-shadow: 0 10px 25px rgba(0,0,0,0.1);">
                 <h2 style="margin: 0 0 1rem 0; font-size: 1.5rem; color: white !important;">🔐 Trial Limit Reached</h2>
                 <p style="margin: 0 0 1.5rem 0; opacity: 0.9; color: white !important;">
-                    You've used your 2 free verifications! This is a demo version on HuggingFace Spaces.
+                    You've used your {TRIAL_LIMIT} free verifications! This is a demo version on HuggingFace Spaces.
                 </p>
                 <p style="margin: 0; font-size: 0.9rem; opacity: 0.8; color: white !important;">
                     For unlimited access, deploy your own instance.
@@ -378,7 +390,7 @@ def verify_news_simple(input_text: str, progress=gr.Progress()) -> str:
         
         # Increment trial count
         new_count = increment_trial_count()
-        remaining = 2 - new_count
+        remaining = TRIAL_LIMIT - new_count
         
         progress(0.05, desc="📝 Processing input text...")
         time.sleep(2)  # 2 sec for input processing
@@ -435,7 +447,7 @@ def verify_news_simple(input_text: str, progress=gr.Progress()) -> str:
         """
         
         return trial_info + main_result + deberta_html
-        
+            
     except Exception as e:
         print(f"ERROR: {e}")
         progress(1.0, desc="❌ Error occurred")
@@ -475,7 +487,7 @@ def create_interface():
         }
         
         /* Button styling - match header section color (#9775fa) */
-        button[variant="primary"], 
+        button[variant="primary"],
         .gr-button-primary,
         .gr-button[data-variant="primary"],
         input[type="submit"] {
@@ -614,7 +626,7 @@ def create_interface():
             <h2 style="margin: 0 0 1rem 0; font-size: 1.4rem; font-weight: 600; color: white !important;">Advanced News Verification System</h2>
             <div style="background: rgba(255,255,255,0.15); border-radius: 8px; padding: 1rem; margin: 1rem 0;">
                 <h3 style="margin: 0 0 0.5rem 0; font-size: 1.2rem; font-weight: bold; color: white !important;">🚀 Free Demo Version</h3>
-                <p style="margin: 0; opacity: 0.9; font-size: 1rem; color: white !important;">Try our news verification system with 2 free searches per session!</p>
+                <p style="margin: 0; opacity: 0.9; font-size: 1rem; color: white !important;">Try our news verification system with 5 free searches per session!</p>
                 <p style="margin: 0.5rem 0 0 0; font-size: 0.9rem; opacity: 0.8; color: white !important;">Powered by DeBERTa AI + Evidence Search</p>
             </div>
         </div>