Update app.py

app.py

@@ -778,8 +778,7 @@ def generate_response_stream(prompt, temperature=0.7, backend=None, max_tokens=2
     final_tokens_per_sec = tokens_generated / elapsed if elapsed > 0 else 0
     yield "", False, final_tokens_per_sec, final_tokens_per_sec, False
 
-
-# PART 3 - Continue from Part 2
+# PART 3 - Continue from Part 2 - FIXED VERSION
 
 if __name__ == "__main__":
     import gradio as gr
@@ -870,9 +869,10 @@ if __name__ == "__main__":
         html += '</div>'
         return html
 
-    user_session = gr.State({})
-
     with gr.Blocks(css=custom_css, theme=gr.themes.Soft(primary_hue="slate")) as demo:
+        # Session state
+        user_session = gr.State({})
+        
         gr.HTML('<div class="announcement-banner">🔐 <strong>SAM-X-1 V3.0 - SECURE ACCESS</strong> 🔐<br>✨ Sign in to unlock AI-powered conversations!<br>🆓 FREE: Nano & Mini unlimited, Fast 10/3h, Large 8/3h<br>⭐ PLUS: Nano/Mini/Fast unlimited, Large 20/3h<br>💎 PRO: Everything unlimited!</div>')
         
         with gr.Tabs() as main_tabs:
@@ -948,11 +948,12 @@ if __name__ == "__main__":
                     submit_upgrade_btn = gr.Button("Submit Request", variant="primary")
                     upgrade_msg = gr.Markdown("")
         
+        # Event handlers
         def handle_login(username, password):
             success, user_data = authenticate_user(username, password)
             if success:
-                return user_data, f"✅ Welcome back, {username}!", "", gr.update(selected=2)
-            return {}, "❌ Invalid credentials!", "", gr.update()
+                return user_data, f"✅ Welcome back, {username}!", ""
+            return {}, "❌ Invalid credentials!", ""
         
         def handle_signup(username, email, password):
             if len(username) < 3:
@@ -1075,19 +1076,23 @@ if __name__ == "__main__":
             return f"{'✅' if success else '❌'} {msg}"
         
         def handle_logout():
-            return {}, "Please sign in", "", gr.update(selected=0)
+            return {}, "Please sign in", ""
         
-        login_btn.click(handle_login, [login_username, login_password], [user_session, login_msg, chat_html, main_tabs])
+        # Wire up events
+        login_btn.click(handle_login, [login_username, login_password], [user_session, login_msg, chat_html])
         signup_btn.click(handle_signup, [signup_username, signup_email, signup_password], [signup_msg])
         chat_tab.select(load_user_interface, [user_session], [user_info, limits_display, admin_panel, model_selector, max_tokens_slider])
+        
         send_outputs = [msg_input, chat_html, speed_display, send_btn, stop_btn]
         send_btn.click(send_message_handler, [msg_input, show_thinking_checkbox, temperature_slider, model_selector, max_tokens_slider, user_session], send_outputs)
         msg_input.submit(send_message_handler, [msg_input, show_thinking_checkbox, temperature_slider, model_selector, max_tokens_slider, user_session], send_outputs)
+        
         stop_btn.click(stop_generation_handler, outputs=[speed_display, send_btn, stop_btn])
         clear_btn.click(clear_chat, outputs=[chat_html, speed_display, send_btn, stop_btn])
         upgrade_btn.click(show_upgrade_panel, outputs=[upgrade_panel])
         submit_upgrade_btn.click(submit_upgrade_request, [upgrade_plan_choice, upgrade_reason, user_session], [upgrade_msg])
-        logout_btn.click(handle_logout, outputs=[user_session, user_info, chat_html, main_tabs])
+        logout_btn.click(handle_logout, outputs=[user_session, user_info, chat_html])
+        
         refresh_users_btn.click(load_all_users, outputs=[users_table])
         refresh_requests_btn.click(load_pending_requests, outputs=[requests_table])
         update_plan_btn.click(admin_update_plan, [admin_username, admin_new_plan], [admin_msg])
@@ -1106,70 +1111,9 @@ if __name__ == "__main__":
         | **Large Model** | 8 msgs/3h | 20 msgs/3h | ✅ Unlimited |
         | **Model Selection** | 🤖 Auto only | ✅ Manual choice | ✅ Manual choice |
         | **Max Tokens** | 256 | 384 | 512 |
-        | **Reset Period** | 3 hours | 3 hours | 3 hours |
-        
-        ### 🎯 How It Works
-        
-        1. **Rolling 3-Hour Window**: Your limits reset 3 hours after you start using the service
-        2. **Australia Timezone**: All times are in Australian Eastern Time (Sydney)
-        3. **Smart Auto-Selection**: FREE users get optimal model selection based on query complexity
-        4. **Flexible Limits**: Nano & Mini are always available for quick questions
-        5. **Upgrade Requests**: Request PLUS or PRO access through the chat interface
-        
-        ### 💡 Tips for FREE Users
-        
-        - **Use Nano & Mini freely** - Perfect for most questions and coding tasks (30-40 tok/s!)
-        - **Save Fast/Large for complex tasks** - You get 10 Fast and 8 Large messages per 3h window
-        - **Let Auto mode decide** - It intelligently picks the right model for your question
-        - **Watch your reset timer** - Plan your usage around the 3-hour window
-        
-        ### 🔐 Admin Access
-        
-        - **Username**: `admin`
-        - **Password**: `admin123`
-        - Admin can manage users, approve upgrades, and reset limits
-        
-        ### 🚀 V3.0 Features
-        
-        - ✅ **User Authentication** - Secure sign in/sign up system
-        - ✅ **Three-Tier Plans** - FREE, PLUS, PRO with different limits
-        - ✅ **Rolling 3-Hour Limits** - Fair usage based on activity
-        - ✅ **Admin Panel** - Full user management and upgrade approval
-        - ✅ **Smart Model Selection** - Auto mode for FREE, manual for PLUS/PRO
-        - ✅ **Usage Tracking** - Monitor your message limits in real-time
-        - ✅ **Australia Timezone** - All times displayed in AEST/AEDT
-        - ✅ **Upgrade Requests** - In-app upgrade request system
-        - ✅ **Per-Model Limits** - Separate counters for Nano, Mini, Fast, Large
-        - ✅ **Dynamic Batching** - Faster UI updates (2-8 token batching)
-        
-        ### 📝 How to Use
-        
-        **Step 1**: Create an account (Sign Up tab)
-        **Step 2**: Sign in with your credentials
-        **Step 3**: Go to Chat tab and start chatting!
-        **Step 4**: Monitor your limits in the limits panel
-        **Step 5**: Request upgrade if you need more access
-        
-        ### 🎓 Example Queries by Plan
-        
-        **FREE Plan (Auto Mode):**
-        - "What is Python?" → Nano (instant, unlimited)
-        - "Write a hello world function" → Mini (fast, unlimited)
-        - "Explain object-oriented programming" → Fast (uses 1 of 10)
-        - "Prove the Pythagorean theorem" → Large (uses 1 of 8)
-        
-        **PLUS/PRO (Manual Selection):**
-        - Choose any model based on your needs
-        - PLUS gets 20 Large messages per 3h
-        - PRO gets unlimited everything!
-        
-        ### 📧 Support
-        
-        For questions or issues, contact the admin through the upgrade request system!
-        
-        ---
         
-        **Made with ❤️ using SAM-X-1 Models**
+        ### 🔐 Default Admin: `admin` / `admin123`
+        ### 🆓 Sign up for FREE account - Nano & Mini unlimited!
         """)
     
-    demo.launch(debug=True, share=True)
+    demo.launch(debug=True, share=False, server_name="0.0.0.0", server_port=7860)