routes bug

app/app.py

@@ -13,211 +13,214 @@ class FlaskApp:
         self.register_routes()
 
     def register_routes(self):
-        @self.flask_app.route('/dashboard', methods=['GET', 'POST'])
-        def user_dashboard():
-            global refresh_needed
-
-            if request.method == 'POST':
-                user_message = request.form.get('message')
-                if user_message:
-                    # User message submission
-                    user_message_dict = {
-                        'content': user_message,
-                        'author': session.get('user_id', 'User'),
-                        'approved': True  # Show immediately but still needs approval in the background
-                    }
-                    self.messages.append(user_message_dict)
-
-                    # Iterate through agents, differentiate between admin and LLM agents
-                    for i, agent_index in enumerate(self.agent_order):
-                        agent = self.agents[agent_index]
-
-                        if agent_index == 0:
-                            # Admin agent, skip LLM request
-                            continue
-
-                        # LLM agent, request response
-                        llm_response = agent.request_llm_response(user_message)
-                        pending_llm_message = {
-                            'content': llm_response,
-                            'author': f'llm{agent_index + 1}',  # Dynamically assign author label
-                            'approved': False
+        if not self.routes_registered:
+            @self.flask_app.route('/dashboard', methods=['GET', 'POST'])
+            def user_dashboard():
+                global refresh_needed
+
+                if request.method == 'POST':
+                    user_message = request.form.get('message')
+                    if user_message:
+                        # User message submission
+                        user_message_dict = {
+                            'content': user_message,
+                            'author': session.get('user_id', 'User'),
+                            'approved': True  # Show immediately but still needs approval in the background
                         }
-                        self.pending_llm_responses.append(pending_llm_message)
-
-                    flash('Message submitted and LLM responses are awaiting approval!')
-                return redirect(url_for('user_dashboard'))
-            
-            refresh_needed = True
-            return render_template('dashboard.html', messages=self.messages, agent_order=self.agent_order)
-
-        @self.flask_app.route('/change_agent_order', methods=['POST'])
-        def change_agent_order():
-            global agent_order
-            # Get the new agent order from the admin form
-            agent_1 = request.form.get('agent_1')
-            agent_2 = request.form.get('agent_2')
-            agent_3 = request.form.get('agent_3')
-
-            # Update the agent order
-            agent_order = [agent_1, agent_2, agent_3]
+                        self.messages.append(user_message_dict)
+
+                        # Iterate through agents, differentiate between admin and LLM agents
+                        for i, agent_index in enumerate(self.agent_order):
+                            agent = self.agents[agent_index]
+
+                            if agent_index == 0:
+                                # Admin agent, skip LLM request
+                                continue
+
+                            # LLM agent, request response
+                            llm_response = agent.request_llm_response(user_message)
+                            pending_llm_message = {
+                                'content': llm_response,
+                                'author': f'llm{agent_index + 1}',  # Dynamically assign author label
+                                'approved': False
+                            }
+                            self.pending_llm_responses.append(pending_llm_message)
+
+                        flash('Message submitted and LLM responses are awaiting approval!')
+                    return redirect(url_for('user_dashboard'))
+                
+                refresh_needed = True
+                return render_template('dashboard.html', messages=self.messages, agent_order=self.agent_order)
+
+            @self.flask_app.route('/change_agent_order', methods=['POST'])
+            def change_agent_order():
+                global agent_order
+                # Get the new agent order from the admin form
+                agent_1 = request.form.get('agent_1')
+                agent_2 = request.form.get('agent_2')
+                agent_3 = request.form.get('agent_3')
+
+                # Update the agent order
+                agent_order = [agent_1, agent_2, agent_3]
+                
+                # Redirect back to the admin dashboard
+                return redirect(url_for('admin_dashboard'))
+
+            @self.flask_app.route('/admin_dashboard', methods=['GET', 'POST'])
+            def admin_dashboard():
+                global refresh_needed
+
+                if request.method == 'POST':
+                    action = request.form.get('action')
+
+                    if action == 'update_system_content':
+                        # Get system content for LLM agents from form input
+                        new_llm_1_content = request.form.get('llm_1_system_content')
+                        new_llm_2_content = request.form.get('llm_2_system_content')
+
+                        # Ensure that the content is updated in the agents
+                        if new_llm_1_content is not None:
+                            self.agents[1].system_content = new_llm_1_content  # Directly update the system_content attribute
+                        if new_llm_2_content is not None:
+                            self.agents[2].system_content = new_llm_2_content  # Directly update the system_content attribute
+
+                        return redirect(url_for('admin_dashboard'))
+                    
+                    if action == 'change_order':
+                        # Capture the selected agent order from the dropdown
+                        self.agent_order = request.form.getlist('agent_order[]')
+
+                        # Refresh the page to see the updated agent order
+                        return redirect("/admin_dashboard")
+
+                    # Admin sending a message to the user
+                    if action == 'send':
+                        admin_message = request.form.get('admin_message')
+                        if admin_message:
+                            new_message = {
+                                'content': admin_message,
+                                'author': 'Admin',  
+                                'approved': True  
+                            }
+                            self.messages.append(new_message)  
+                            flash('Message sent to the user!')
+                        else:
+                            flash('Error: Message content is empty.')
+                        return redirect(url_for('admin_dashboard'))
             
-            # Redirect back to the admin dashboard
-            return redirect(url_for('admin_dashboard'))
-
-        @self.flask_app.route('/admin_dashboard', methods=['GET', 'POST'])
-        def admin_dashboard():
-            global refresh_needed
-
-            if request.method == 'POST':
-                action = request.form.get('action')
-
-                if action == 'update_system_content':
-                    # Get system content for LLM agents from form input
-                    new_llm_1_content = request.form.get('llm_1_system_content')
-                    new_llm_2_content = request.form.get('llm_2_system_content')
-
-                    # Ensure that the content is updated in the agents
-                    if new_llm_1_content is not None:
-                        self.agents[1].system_content = new_llm_1_content  # Directly update the system_content attribute
-                    if new_llm_2_content is not None:
-                        self.agents[2].system_content = new_llm_2_content  # Directly update the system_content attribute
-
-                    return redirect(url_for('admin_dashboard'))
+                    message_id = request.form.get('message_id')
+
+                    if message_id is not None:
+                        message_id = int(message_id)
+
+                    # Approve action
+                    if action == 'approve':
+                        if message_id is not None and message_id < len(self.pending_llm_responses):
+                            self.pending_llm_responses[message_id]['approved'] = True
+                            self.messages.append(self.pending_llm_responses.pop(message_id))
+
+                        return redirect(url_for('admin_dashboard'))
+
+                    # Delete action
+                    if action == 'delete':
+                        if message_id is not None and message_id < len(self.pending_llm_responses):
+                            del self.pending_llm_responses[message_id]
+                        return redirect(url_for('admin_dashboard'))
+
+                    # Regenerate action
+                    if action == 'regenerate':
+                        if message_id is not None and message_id < len(self.pending_llm_responses):
+                            message_content = self.pending_llm_responses[message_id]['content']
+                            
+                            for agent in self.agents:
+                                if agent is not None:
+                                    regenerated_response = agent.request_llm_response(message_content)
+                                    self.pending_llm_responses[message_id]['content'] = regenerated_response
+                                    break
+                        return redirect(url_for('admin_dashboard'))
                 
-                if action == 'change_order':
-                    # Capture the selected agent order from the dropdown
-                    self.agent_order = request.form.getlist('agent_order[]')
-
-                    # Refresh the page to see the updated agent order
-                    return redirect("/admin_dashboard")
-
-                # Admin sending a message to the user
-                if action == 'send':
-                    admin_message = request.form.get('admin_message')
-                    if admin_message:
-                        new_message = {
-                            'content': admin_message,
-                            'author': 'Admin',  
-                            'approved': True  
-                        }
-                        self.messages.append(new_message)  
-                        flash('Message sent to the user!')
+                # Set the refresh_needed flag to True
+                refresh_needed = True
+
+                return render_template('admin_dashboard.html',
+                                    all_messages=self.messages,
+                                    pending_messages=self.pending_llm_responses,
+                                    agent_order=self.agent_order,
+                                    llm_1_system_content=self.agents[1].system_content,
+                                    llm_2_system_content=self.agents[2].system_content)
+
+            @self.flask_app.route('/reinitialize', methods=['POST'])
+            def reinitialize_dashboards():
+                # Reset session data or other necessary variables
+                session.clear()  # Clears the session, you can also reset other custom variables here if needed
+
+                # Reinitialize your variables here (like message history, agent order, etc.)
+                # Assuming self.agent_order and messages are variables to reset
+                self.messages = []  # Reset message history
+                self.pending_llm_responses = []  # Reset pending LLM messages
+                self.agent_order = [0, 1, 2]  # Reset agent order to default
+                
+                # Redirect to the dashboard (or admin dashboard)
+                return redirect(url_for('admin_dashboard'))  # You can redirect to 'user_dashboard' or another page if necessary
+            
+            @self.flask_app.route('/user_dashboard_reset', methods=['POST'])
+            def user_dashboard_reset():
+                # After reinitializing, update the user dashboard
+                return render_template('partials/_user_message_list.html', messages=self.messages)
+
+            @self.flask_app.route('/login', methods=['GET', 'POST'])
+            def login():
+                if request.method == 'POST':
+                    username = request.form.get('username')
+                    
+                    if username == 'admin':
+                        session['username'] = 'admin'
+                        return redirect(url_for('admin_dashboard'))
+                    elif username == 'user':
+                        session['username'] = 'user'
+                        return redirect(url_for('user_dashboard'))  # Use the correct route name
                     else:
-                        flash('Error: Message content is empty.')
-                    return redirect(url_for('admin_dashboard'))
-        
-                message_id = request.form.get('message_id')
-
-                if message_id is not None:
-                    message_id = int(message_id)
-
-                # Approve action
-                if action == 'approve':
-                    if message_id is not None and message_id < len(self.pending_llm_responses):
-                        self.pending_llm_responses[message_id]['approved'] = True
-                        self.messages.append(self.pending_llm_responses.pop(message_id))
-
-                    return redirect(url_for('admin_dashboard'))
-
-                # Delete action
-                if action == 'delete':
-                    if message_id is not None and message_id < len(self.pending_llm_responses):
-                        del self.pending_llm_responses[message_id]
-                    return redirect(url_for('admin_dashboard'))
-
-                # Regenerate action
-                if action == 'regenerate':
-                    if message_id is not None and message_id < len(self.pending_llm_responses):
-                        message_content = self.pending_llm_responses[message_id]['content']
-                        
-                        for agent in self.agents:
-                            if agent is not None:
-                                regenerated_response = agent.request_llm_response(message_content)
-                                self.pending_llm_responses[message_id]['content'] = regenerated_response
-                                break
-                    return redirect(url_for('admin_dashboard'))
+                        # If the username is incorrect, show an error message
+                        return "Invalid username. Please use 'user' or 'admin'."
+                return render_template('login.html')
+
+            @self.flask_app.route('/logout')
+            def logout():
+                session.clear()
+                return redirect(url_for('login'))  
+
+            # Route for index.html
+            @self.flask_app.route('/')
+            def index():
+                return render_template('index.html') 
+
+            # Endpoint to check whether a refresh is needed
+            @self.flask_app.route('/check_refresh')
+            def check_refresh():
+                global refresh_needed
+                return jsonify({"refresh_needed": refresh_needed})
+
+            # Reset the refresh flag
+            @self.flask_app.route('/reset_refresh', methods=['POST'])
+            def reset_refresh():
+                global refresh_needed
+                refresh_needed = False
+                return '', 204
+
+            @self.flask_app.route('/get_message_history', methods=['GET'])
+            def get_message_history():
+                return jsonify(self.messages)
+
+            @self.flask_app.route('/get_pending_messages', methods=['GET'])
+            def get_pending_messages():
+                return jsonify(self.pending_llm_responses)
             
-            # Set the refresh_needed flag to True
-            refresh_needed = True
-
-            return render_template('admin_dashboard.html',
-                                all_messages=self.messages,
-                                pending_messages=self.pending_llm_responses,
-                                agent_order=self.agent_order,
-                                llm_1_system_content=self.agents[1].system_content,
-                                llm_2_system_content=self.agents[2].system_content)
-
-        @self.flask_app.route('/reinitialize', methods=['POST'])
-        def reinitialize_dashboards():
-            # Reset session data or other necessary variables
-            session.clear()  # Clears the session, you can also reset other custom variables here if needed
-
-            # Reinitialize your variables here (like message history, agent order, etc.)
-            # Assuming self.agent_order and messages are variables to reset
-            self.messages = []  # Reset message history
-            self.pending_llm_responses = []  # Reset pending LLM messages
-            self.agent_order = [0, 1, 2]  # Reset agent order to default
+            @self.flask_app.route('/get_agent_order', methods=['GET'])
+            def get_agent_order():
+                global agent_order
+                return jsonify({"agent_order": agent_order})
             
-            # Redirect to the dashboard (or admin dashboard)
-            return redirect(url_for('admin_dashboard'))  # You can redirect to 'user_dashboard' or another page if necessary
-        
-        @self.flask_app.route('/user_dashboard_reset', methods=['POST'])
-        def user_dashboard_reset():
-            # After reinitializing, update the user dashboard
-            return render_template('partials/_user_message_list.html', messages=self.messages)
-
-        @self.flask_app.route('/login', methods=['GET', 'POST'])
-        def login():
-            if request.method == 'POST':
-                username = request.form.get('username')
-                
-                if username == 'admin':
-                    session['username'] = 'admin'
-                    return redirect(url_for('admin_dashboard'))
-                elif username == 'user':
-                    session['username'] = 'user'
-                    return redirect(url_for('user_dashboard'))  # Use the correct route name
-                else:
-                    # If the username is incorrect, show an error message
-                    return "Invalid username. Please use 'user' or 'admin'."
-            return render_template('login.html')
-
-        @self.flask_app.route('/logout')
-        def logout():
-            session.clear()
-            return redirect(url_for('login'))  
-
-        # Route for index.html
-        @self.flask_app.route('/')
-        def index():
-            return render_template('index.html') 
-
-        # Endpoint to check whether a refresh is needed
-        @self.flask_app.route('/check_refresh')
-        def check_refresh():
-            global refresh_needed
-            return jsonify({"refresh_needed": refresh_needed})
-
-        # Reset the refresh flag
-        @self.flask_app.route('/reset_refresh', methods=['POST'])
-        def reset_refresh():
-            global refresh_needed
-            refresh_needed = False
-            return '', 204
-
-        @self.flask_app.route('/get_message_history', methods=['GET'])
-        def get_message_history():
-            return jsonify(self.messages)
-
-        @self.flask_app.route('/get_pending_messages', methods=['GET'])
-        def get_pending_messages():
-            return jsonify(self.pending_llm_responses)
-        
-        @self.flask_app.route('/get_agent_order', methods=['GET'])
-        def get_agent_order():
-            global agent_order
-            return jsonify({"agent_order": agent_order})
+        self.routes_registered = True  # Mark routes as registered
 
 if __name__ == '__main__':
     # Initialize the LLM agents with default system content