Update app.py

app.py

@@ -21,7 +21,6 @@ def check_paywall(user_id: str):
         if data.get("connected"):
             return True, ""
         else:
-            # Provide user the URL to connect their account or top up
             message = f"Please connect or top up to access. Visit: {data.get('url', 'No URL provided')}"
             return False, message
     except Exception as e:
@@ -60,22 +59,27 @@ def chat_completion(user_id, system_message, history, user_message, max_tokens,
         print(f"[Paywall] Chat completion error: {e}")
         return f"Error retrieving completion: {e}"
 
-# Full respond function used by Gradio chatbot interface
-def respond(message, history, system_message, max_tokens, temperature, top_p, hf_token):
-    user_id = "demo_user"  # Replace or dynamically assign per user session
+# Updated respond function fit for Gradio ChatInterface
+def respond(
+    message, history, system_message, max_tokens, temperature, top_p, hf_token
+):
+    user_id = "demo_user"  # Replace with actual user session ID
+
+    # Check paywall connection
     connected, paywall_msg = check_paywall(user_id)
     
     if not connected:
-        # User not connected. Show connection URL or message.
-        return (history, paywall_msg)
+        # When user is not connected, return empty history and paywall message string
+        return [], paywall_msg
     
-    # User connected - forward chat message to paywalls chat completions endpoint
+    # If connected, get chatbot reply
     reply_text = chat_completion(user_id, system_message, history, message, max_tokens, temperature, top_p)
     
-    history = history + [{"role": "user", "content": message}, {"role": "assistant", "content": reply_text}]
-    return history, ""
+    # Append new exchange to history
+    new_history = history + [{"role": "user", "content": message}, {"role": "assistant", "content": reply_text}]
+    return new_history, ""  # second return clears the input box
 
-# Gradio interface setup
+# Gradio UI setup with additional inputs
 chatbot = gr.ChatInterface(
     respond,
     type="messages",