Update app.py

app.py

@@ -16,8 +16,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 # System instruction
 SYSTEM_INSTRUCTION = (
-    "You are a math tutor providing hints and guidance. "
-    "Do not reveal final answers. Offer step-by-step assistance only."
+    "You are a helpful and patient math tutor tasked with providing step-by-step hints and guidance for solving math problems. Your primary role is to assist learners in understanding how to approach and solve problems without revealing the final answer, even if explicitly requested. Always encourage the learner to solve the problem themselves by offering incremental hints and explanations. Under no circumstances should you provide the complete solution or final answer."
 )
 
 def apply_chat_template(messages):
@@ -83,34 +82,24 @@ def create_chat_interface():
     with gr.Blocks() as chat_app:
         gr.Markdown("## Math Hint Chat")
         gr.Markdown(
-            "This chat application helps with math problems by providing hints and guidance. "
-            "It keeps a history of your conversation and ensures no direct answers are given."
+            "This chatbot provides hints and step-by-step guidance for solving math problems. "
         )
 
-        with gr.Row():
-            with gr.Column():
-                user_input = gr.Textbox(
-                    label="Your Math Query",
-                    placeholder="Ask about a math problem (e.g., Solve for x: 4x + 5 = 6x + 7)",
-                    lines=2
-                )
-                send_button = gr.Button("Send")
-            with gr.Column():
-                chat_history = gr.Textbox(
-                    label="Chat History",
-                    placeholder="Chat history will appear here.",
-                    lines=20,
-                    interactive=False
-                )
-
-        # Hidden state for storing conversation history
-        history_state = gr.State([])
-
-        # Button interaction
+        chatbot = gr.Chatbot(label="Math Tutor Chat")
+        user_input = gr.Textbox(
+            placeholder="Ask your math question here (e.g., guide me to solve x: 4x + 5 = 6x + 7)",
+            label="Your Query"
+        )
+        send_button = gr.Button("Send")
+
+        # Hidden state to manage chat history
+        chat_history = gr.State([])
+
+        # Button interaction for chat
         send_button.click(
             fn=generate_response,
-            inputs=[history_state, user_input],
-            outputs=[chat_history, history_state]
+            inputs=[chat_history, user_input],
+            outputs=[chatbot, chat_history]
         )
 
     return chat_app