Update app.py

app.py

@@ -7,7 +7,6 @@ from datetime import datetime
 # Load secrets
 PAYWALLS_API_KEY = os.environ.get("PAYWALLS_API_KEY")
 PAYWALLS_API_URL = os.environ.get("PAYWALLS_API_URL", "https://api.paywalls.ai/v1")
-HF_MODEL = "openai/gpt-oss-20b"
 
 # Free tier limits
 FREE_MAX_TOKENS = 150
@@ -23,80 +22,102 @@ STYLE_PRESETS = {
 }
 
 def check_paywall(user_id: str):
+    """Check if user has active premium subscription"""
     headers = {
-        "Authorization": f"Bearer {PAYWALLS_API_KEY}",
-        "X-Paywall-User": user_id
+        "Authorization": f"Bearer {PAYWALLS_API_KEY}"
     }
     try:
-        resp = requests.get(f"{PAYWALLS_API_URL}/user/connect?user={user_id}", headers=headers, timeout=10)
+        resp = requests.get(f"{PAYWALLS_API_URL}/users/{user_id}", headers=headers, timeout=10)
         resp.raise_for_status()
         data = resp.json()
-        return data.get("connected", False), data.get('url', '')
+        # Check if user has active subscription
+        is_connected = data.get("status") == "active" or data.get("connected", False)
+        paywall_url = f"https://paywalls.ai/connect?api_key={PAYWALLS_API_KEY}&user={user_id}"
+        return is_connected, paywall_url
+    except Exception as e:
+        print(f"[Paywall] Check error: {e}")
+        return False, f"https://paywalls.ai/connect?api_key={PAYWALLS_API_KEY}&user={user_id}"
+
+def free_tier_generation(system_message, history, user_message, max_tokens, temperature):
+    """Free tier using Hugging Face Inference API"""
+    try:
+        from huggingface_hub import InferenceClient
+        client = InferenceClient()
+        messages = [{"role": "system", "content": system_message}] + history + [{"role": "user", "content": user_message}]
+        
+        reply = client.chat_completion(
+            messages=messages,
+            max_tokens=max_tokens,
+            temperature=temperature
+        ).choices[0].message.content
+        
+        return reply
     except Exception as e:
-        print(f"[Paywall] Error: {e}")
-        return False, ""
+        print(f"[Free Tier] Error: {e}")
+        return f"⚠️ Free tier service error: {str(e)}"
 
-def chat_completion(user_id, system_message, history, user_message, max_tokens, temperature, top_p):
+def premium_generation(user_id, system_message, history, user_message, max_tokens, temperature, top_p):
+    """Premium tier using Paywalls proxy"""
     headers = {
         "Authorization": f"Bearer {PAYWALLS_API_KEY}",
-        "Content-Type": "application/json",
-        "X-Paywall-User": user_id
+        "Content-Type": "application/json"
     }
     
     messages = [{"role": "system", "content": system_message}] + history + [{"role": "user", "content": user_message}]
+    
+    # Using Paywalls proxy endpoint - adjust model based on your Paywalls setup
     payload = {
-        "model": HF_MODEL,
+        "model": "gpt-3.5-turbo",  # Use model available in your Paywalls account
         "messages": messages,
-        "stream": False,
-        "user": user_id,
         "max_tokens": max_tokens,
         "temperature": temperature,
-        "top_p": top_p
+        "top_p": top_p,
+        "user": user_id
     }
     
     try:
-        response = requests.post(f"{PAYWALLS_API_URL}/chat/completions", headers=headers, json=payload, timeout=60)
+        response = requests.post(
+            f"{PAYWALLS_API_URL}/chat/completions",
+            headers=headers,
+            json=payload,
+            timeout=60
+        )
         response.raise_for_status()
         data = response.json()
-        choices = data.get("choices", [])
-        if choices:
-            return choices[0]["message"]["content"]
-        return "No response from model."
+        
+        if data.get("choices"):
+            return data["choices"][0]["message"]["content"]
+        return "No response from premium model."
+        
+    except requests.exceptions.HTTPError as e:
+        print(f"[Premium] HTTP Error: {e.response.status_code} - {e.response.text}")
+        return f"⚠️ Premium service error: {e.response.status_code}"
     except Exception as e:
-        print(f"[Paywall] Error: {e}")
-        return f"Error: {e}"
+        print(f"[Premium] Error: {e}")
+        return f"⚠️ Premium service error: {str(e)}"
 
 def respond(message, history, style, max_tokens, temperature, top_p):
-    user_id = "demo_user"
+    user_id = "demo_user"  # Replace with actual user session
     is_premium, paywall_url = check_paywall(user_id)
     
+    # Check style restriction for free users
     if not is_premium:
         max_tokens = min(max_tokens, FREE_MAX_TOKENS)
         if style != "neutral":
             return history + [{"role": "assistant", "content": f"🔒 Style customization requires premium. [Unlock here]({paywall_url})"}]
-        
-        # FREE TIER: Use direct Hugging Face API
-        try:
-            from huggingface_hub import InferenceClient
-            client = InferenceClient()
-            reply = client.chat_completion(
-                messages=[{"role": "system", "content": STYLE_PRESETS["neutral"]}] + history + [{"role": "user", "content": message}],
-                max_tokens=max_tokens,
-                temperature=temperature
-            ).choices[0].message.content
-            return history + [{"role": "user", "content": message}, {"role": "assistant", "content": f"{reply}\n\n---\n_🆓 FREE_"}]
-        except Exception as e:
-            print(f"[Free Tier] Error: {e}")
-            return history + [{"role": "assistant", "content": f"⚠️ Service temporarily unavailable. Error: {e}"}]
     
-    # PREMIUM: Use Paywalls API
     system_message = STYLE_PRESETS.get(style, STYLE_PRESETS["neutral"])
-    reply = chat_completion(user_id, system_message, history, message, max_tokens, temperature, top_p)
     
-    if "Error:" in reply:
-        return history + [{"role": "assistant", "content": f"⚠️ {reply}"}]
+    # Generate response based on tier
+    if is_premium:
+        reply = premium_generation(user_id, system_message, history, message, max_tokens, temperature, top_p)
+        tier_badge = "✨ PREMIUM"
+    else:
+        reply = free_tier_generation(system_message, history, message, max_tokens, temperature)
+        tier_badge = "🆓 FREE"
     
-    return history + [{"role": "user", "content": message}, {"role": "assistant", "content": f"{reply}\n\n---\n_✨ PREMIUM_"}]
+    full_reply = f"{reply}\n\n---\n_{tier_badge}_"
+    return history + [{"role": "user", "content": message}, {"role": "assistant", "content": full_reply}]
 
 def export_conversation(history):
     user_id = "demo_user"
@@ -148,5 +169,4 @@ with gr.Blocks(title="Creative Writing Assistant") as demo:
     export_btn.click(export_conversation, [chatbot], [download_file, export_status])
 
 if __name__ == "__main__":
-    demo.launch()
-```
+    demo.launch()