Update app.py

app.py

@@ -4,7 +4,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
 import torch
 from threading import Thread
 import re
+import uuid
 
+# Load model and tokenizer
 phi4_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -16,7 +18,15 @@ def format_math(text):
     text = text.replace(r"\(", "$").replace(r"\)", "$")
     return text
 
+# Global dictionary to store all conversations
+conversations = {}
+
+# Function to generate a unique conversation ID
+def generate_conversation_id():
+    return str(uuid.uuid4())[:8]
+
 @spaces.GPU(duration=60)
+# Function to generate response
 def generate_response(user_message, max_tokens, temperature, top_p, history_state):
     if not user_message.strip():
         return history_state, history_state
@@ -54,7 +64,7 @@ def generate_response(user_message, max_tokens, temperature, top_p, history_stat
 
     try:
         thread = Thread(target=model.generate, kwargs=generation_kwargs)
-        thread.start() 
+        thread.start()
     except Exception:
         yield history_state + [{"role": "user", "content": user_message}, {"role": "assistant", "content": "⚠️ Generation failed."}], history_state
         return
@@ -78,25 +88,35 @@ def generate_response(user_message, max_tokens, temperature, top_p, history_stat
 
     yield new_history, new_history
 
+# Example messages
 example_messages = {
     "JEE Main 2025 Combinatorics": "From all the English alphabets, five letters are chosen and are arranged in alphabetical order. The total number of ways, in which the middle letter is 'M', is?",
     "JEE Main 2025 Co-ordinate Geometry": "A circle \\(C\\) of radius 2 lies in the second quadrant and touches both the coordinate axes. Let \\(r\\) be the radius of a circle that has centre at the point \\((2, 5)\\) and intersects the circle \\(C\\) at exactly two points. If the set of all possible values of \\(r\\) is the interval \\((\\alpha, \\beta)\\), then \\(3\\beta - 2\\alpha\\) is?",
-    "JEE Main 2025 Prob-Stats": "A coin is tossed three times. Let \(X\) denote the number of times a tail follows a head. If \\(\\mu\\) and \\(\\sigma^2\\) denote the mean and variance of \\(X\\), then the value of \\(64(\\mu + \\sigma^2)\\) is?"
+    "JEE Main 2025 Prob-Stats": "A coin is tossed three times. Let \(X\) denote the number of times a tail follows a head. If \\(\\mu\\) and \\(\\sigma^2\\) denote the mean and variance of \\(X\\), then the value of \\(64(\\mu + \\sigma^2)\\) is?",
+    "JEE Main 2025 Physics": "A massless spring gets elongated by amount x_1 under a tension of 5 N . Its elongation is x_2 under the tension of 7 N . For the elongation of 5x_1 - 2x_2 , the tension in the spring will be?"
 }
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # Ramanujan Ganit R1 14B V1 Chatbot 
-    
+
         Welcome to the Ramanujan Ganit R1 14B V1 Chatbot, developed by Fractal AI Research! 
-        
+
         Our model excels at reasoning tasks in mathematics and science.
-        
+
         Try the example problems below from JEE Main 2025 or type in your own problems to see how our model breaks down complex reasoning problems.
         """
     )
 
+    # Sidebar for conversation history
+    with gr.Sidebar():
+        gr.Markdown("## Conversations")
+        conversation_selector = gr.Radio(choices=[], label="Select Conversation", interactive=True)
+        new_convo_button = gr.Button("New Conversation")
+
+    # State to store current conversation ID and history
+    current_convo_id = gr.State(generate_conversation_id())
     history_state = gr.State([])
 
     with gr.Row():
@@ -125,24 +145,51 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(label="Chat", type="messages")
-            with gr.Column():
+            with gr.Row():
                 user_input = gr.Textbox(
                     label="User Input",
                     placeholder="Type your question here...",
-                    lines=2
+                    scale=8  # This makes the textbox take up the entire width
                 )
-                with gr.Row():
-                    submit_button = gr.Button("Send", variant="primary")
-                    clear_button = gr.Button("Clear")
+                with gr.Column():
+                    submit_button = gr.Button("Send", variant="primary", scale=1)
+                    clear_button = gr.Button("Clear", scale=1)
             gr.Markdown("**Try these examples:**")
             with gr.Row():
                 example1_button = gr.Button("JEE Main 2025 Combinatorics")
                 example2_button = gr.Button("JEE Main 2025 Co-ordinate Geometry")
                 example3_button = gr.Button("JEE Main 2025 Prob-Stats")
-
+                example4_button = gr.Button("JEE Main 2025 Physics")
+
+    # Function to update conversation list
+    def update_conversation_list():
+        return list(conversations.keys())
+
+    # Function to start a new conversation
+    def start_new_conversation():
+        new_id = generate_conversation_id()
+        conversations[new_id] = []
+        return new_id, [], gr.update(choices=update_conversation_list(), value=new_id)
+
+    # Function to load selected conversation
+    def load_conversation(selected_id):
+        if selected_id in conversations:
+            return selected_id, conversations[selected_id], conversations[selected_id]
+        else:
+            return current_convo_id.value, history_state.value, history_state.value
+
+    # Send message
+    def send_message(user_message, max_tokens, temperature, top_p, convo_id, history):
+        if convo_id not in conversations:
+            conversations[convo_id] = history
+        for updated_history, new_history in generate_response(user_message, max_tokens, temperature, top_p, history):
+            conversations[convo_id] = new_history
+            yield updated_history, new_history
+
+    # Button and event handlers
     submit_button.click(
-        fn=generate_response,
-        inputs=[user_input, max_tokens_slider, temperature_slider, top_p_slider, history_state],
+        fn=send_message,
+        inputs=[user_input, max_tokens_slider, temperature_slider, top_p_slider, current_convo_id, history_state],
         outputs=[chatbot, history_state]
     ).then(
         fn=lambda: gr.update(value=""),
@@ -156,6 +203,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[chatbot, history_state]
     )
 
+    new_convo_button.click(
+        fn=start_new_conversation,
+        inputs=None,
+        outputs=[current_convo_id, history_state, conversation_selector]
+    )
+
+    conversation_selector.change(
+        fn=load_conversation,
+        inputs=conversation_selector,
+        outputs=[current_convo_id, history_state, chatbot]
+    )
+
     example1_button.click(
         fn=lambda: gr.update(value=example_messages["JEE Main 2025 Combinatorics"]),
         inputs=None,
@@ -171,5 +230,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         inputs=None,
         outputs=user_input
     )
+    example4_button.click(
+        fn=lambda: gr.update(value=example_messages["JEE Main 2025 Physics"]),
+        inputs=None,
+        outputs=user_input
+    )
 
 demo.launch(share=True, ssr_mode=False)