Update app.py

app.py

@@ -18,21 +18,11 @@ def format_math(text):
     text = text.replace(r"\(", "$").replace(r"\)", "$")
     return text
 
-# Global dictionary to store all conversations
+# Global dictionary to store all conversations: {id: {"title": str, "messages": list}}
 conversations = {}
 
-# Function to generate a user-friendly conversation title
-def generate_conversation_name(user_message):
-    words = user_message.strip().split()
-    if not words:
-        return "Conversation-" + str(uuid.uuid4())[:8]
-    name = " ".join(words[:5])
-    base_name = name
-    suffix = 1
-    while name in conversations:
-        name = f"{base_name} ({suffix})"
-        suffix += 1
-    return name
+def generate_conversation_id():
+    return str(uuid.uuid4())[:8]
 
 @spaces.GPU(duration=60)
 def generate_response(user_message, max_tokens, temperature, top_p, history_state):
@@ -104,23 +94,33 @@ example_messages = {
 }
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Ramanujan Ganit R1 14B V1 Chatbot\n\nWelcome to the Ramanujan Ganit R1 14B V1 Chatbot, developed by Fractal AI Research!\n\nOur model excels at reasoning tasks in mathematics and science.\n\nTry the example problems below from JEE Main 2025 or type in your own problems to see how our model breaks down complex reasoning problems.")
+    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.
+        """
+    )
 
     with gr.Sidebar():
         gr.Markdown("## Conversations")
         conversation_selector = gr.Radio(choices=[], label="Select Conversation", interactive=True)
         new_convo_button = gr.Button("New Conversation")
 
-    current_convo_id = gr.State("")
+    current_convo_id = gr.State(generate_conversation_id())
     history_state = gr.State([])
 
     with gr.Row():
         with gr.Column(scale=1):
             gr.Markdown("### Settings")
-            max_tokens_slider = gr.Slider(6144, 32768, step=1024, value=16384, label="Max Tokens")
+            max_tokens_slider = gr.Slider(minimum=6144, maximum=32768, step=1024, value=16384, label="Max Tokens")
             with gr.Accordion("Advanced Settings", open=False):
-                temperature_slider = gr.Slider(0.1, 2.0, value=0.6, label="Temperature")
-                top_p_slider = gr.Slider(0.1, 1.0, value=0.95, label="Top-p")
+                temperature_slider = gr.Slider(minimum=0.1, maximum=2.0, value=0.6, label="Temperature")
+                top_p_slider = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, label="Top-p")
 
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(label="Chat", type="messages")
@@ -137,33 +137,33 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 example4_button = gr.Button("JEE Main 2025 Physics")
 
     def update_conversation_list():
-        return list(conversations.keys())
+        return [conversations[cid]["title"] for cid in conversations]
 
     def start_new_conversation():
-        return "", [], gr.update(choices=update_conversation_list(), value=None)
+        new_id = generate_conversation_id()
+        conversations[new_id] = {"title": f"New Conversation {new_id}", "messages": []}
+        return new_id, [], gr.update(choices=update_conversation_list(), value=conversations[new_id]["title"])
 
-    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
+    def load_conversation(selected_title):
+        for cid, convo in conversations.items():
+            if convo["title"] == selected_title:
+                return cid, convo["messages"], convo["messages"]
+        return current_convo_id.value, history_state.value, history_state.value
 
     def send_message(user_message, max_tokens, temperature, top_p, convo_id, history):
-        new_convo = False
-        if not convo_id:
-            new_convo = True
-            convo_id = generate_conversation_name(user_message)
-            conversations[convo_id] = []
-
+        if convo_id not in conversations:
+            title = user_message.strip().split("\n")[0][:40]
+            conversations[convo_id] = {"title": title, "messages": history}
+        if conversations[convo_id]["title"].startswith("New Conversation"):
+            conversations[convo_id]["title"] = user_message.strip().split("\n")[0][:40]
         for updated_history, new_history in generate_response(user_message, max_tokens, temperature, top_p, history):
-            conversations[convo_id] = new_history
-            updates = [gr.update(choices=update_conversation_list(), value=convo_id)] if new_convo else []
-            yield updated_history, new_history, convo_id, *updates
+            conversations[convo_id]["messages"] = new_history
+            yield updated_history, new_history, gr.update(choices=update_conversation_list(), value=conversations[convo_id]["title"])
 
     submit_button.click(
         fn=send_message,
         inputs=[user_input, max_tokens_slider, temperature_slider, top_p_slider, current_convo_id, history_state],
-        outputs=[chatbot, history_state, current_convo_id, conversation_selector]
+        outputs=[chatbot, history_state, conversation_selector]
     ).then(
         fn=lambda: gr.update(value=""),
         inputs=None,
@@ -189,8 +189,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
     example1_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Combinatorics"]), inputs=None, outputs=user_input)
-    example2_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Co-ordinate Geometry"]), inputs=None, outputs=user_input)
-    example3_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Prob-Stats"]), inputs=None, outputs=user_input)
-    example4_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Physics"]), inputs=None, outputs=user_input)
+    example2_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Coordinate Geometry"]), inputs=None, outputs=user_input)
+    example3_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Probability & Statistics"]), inputs=None, outputs=user_input)
+    example4_button.click(fn=lambda: gr.update(value=example_messages["JEE Main 2025 Laws of Motion"]), inputs=None, outputs=user_input)
 
 demo.launch(share=True, ssr_mode=False)