Update app.py

app.py

@@ -67,21 +67,21 @@ def chat_completion(user_id, system_message, history, user_message, max_tokens,
         return f"Error: {e}"
 
 def respond(message, history, style, max_tokens, temperature, top_p):
-    user_id = "demo_user"  # Replace with gr.Request for real user sessions
+    user_id = "demo_user"
     is_premium, paywall_url = check_paywall(user_id)
     
-    # Apply free tier limits
     if not is_premium:
         max_tokens = min(max_tokens, FREE_MAX_TOKENS)
         if style != "neutral":
-            return f"🔒 Style customization requires premium. [Unlock here]({paywall_url})"
+            return history + [{"role": "assistant", "content": f"🔒 Style customization requires premium. [Unlock here]({paywall_url})"}]
     
     system_message = STYLE_PRESETS.get(style, STYLE_PRESETS["neutral"])
     reply = chat_completion(user_id, system_message, history, message, max_tokens, temperature, top_p)
     
-    # Add tier indicator
     tier_badge = "✨ PREMIUM" if is_premium else "🆓 FREE"
-    return f"{reply}\n\n---\n_{tier_badge}_"
+    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"
@@ -90,7 +90,6 @@ def export_conversation(history):
     if not is_premium:
         return None, f"🔒 Export requires premium. [Unlock here]({paywall_url})"
     
-    # Generate text file
     content = f"Creative Writing Session - {datetime.now().strftime('%Y-%m-%d %H:%M')}\n\n"
     for msg in history:
         role = msg["role"].upper()
@@ -129,8 +128,8 @@ with gr.Blocks(title="Creative Writing Assistant") as demo:
             export_status = gr.Textbox(label="Status", interactive=False)
     
     # Event handlers
-    submit.click(respond, [msg, chatbot, style, max_tokens, temperature, top_p], [chatbot])
-    msg.submit(respond, [msg, chatbot, style, max_tokens, temperature, top_p], [chatbot])
+    submit.click(respond, [msg, chatbot, style, max_tokens, temperature, top_p], [chatbot]).then(lambda: "", None, [msg])
+    msg.submit(respond, [msg, chatbot, style, max_tokens, temperature, top_p], [chatbot]).then(lambda: "", None, [msg])
     export_btn.click(export_conversation, [chatbot], [download_file, export_status])
 
 if __name__ == "__main__":