Update app.py

app.py

@@ -12,8 +12,8 @@ def load_resources():
     load_dotenv()
     huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
     subprocess.run(["huggingface-cli", "login", "--token", huggingface_token], capture_output=True)
-    tokenizer = AutoTokenizer.from_pretrained("istiak101/TinyLlama-1.1B-Finetuned")
-    model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Finetuned")
+    tokenizer = AutoTokenizer.from_pretrained("istiak101/TinyLlama-1.1B-Chat-v0.6-rag-finetunedv1.0")
+    model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Chat-v0.6-rag-finetunedv1.0")
     return model, tokenizer
 
 def create_test_prompt(question, context, tokenizer):
@@ -169,17 +169,20 @@ else:
     st.sidebar.write("No conversations yet. Start one below!")
 
 # --- New Conversation ---
-new_topic = st.sidebar.text_input("New Conversation Name")
-if st.sidebar.button("Start New Conversation"):
-    if new_topic.strip() and new_topic not in st.session_state.chat_sessions:
-        st.session_state.chat_sessions[new_topic] = []
-        st.session_state.current_conversation = new_topic
-        st.sidebar.success(f"Started new conversation: {new_topic}")
-        st.rerun()
-    elif not new_topic.strip():
-        st.sidebar.warning("Please enter a name.")
-    else:
-        st.sidebar.warning("Conversation already exists!")
+with st.sidebar.form(key='new_conversation_form', clear_on_submit=True):
+    new_topic = st.text_input("New Conversation Name")
+    submit_button = st.form_submit_button("Start New Conversation")
+    
+    if submit_button:
+        if new_topic.strip() and new_topic not in st.session_state.chat_sessions:
+            st.session_state.chat_sessions[new_topic] = []
+            st.session_state.current_conversation = new_topic
+            st.sidebar.success(f"Started new conversation: {new_topic}")
+            st.rerun()
+        elif not new_topic.strip():
+            st.sidebar.warning("Please enter a name.")
+        else:
+            st.sidebar.warning("Conversation already exists!")
 
 # --- Main Chat Area ---
 if st.session_state.current_conversation:
@@ -192,7 +195,7 @@ if st.session_state.current_conversation:
             if msg["role"] == "user":
                 if st.session_state.edit_mode.get(idx, False):
                     # Split the message into question and context
-                    question_input, context_input = msg["text"].split("<br>")
+                    question_input, context_input = msg["text"].split("<br><br>")
                     # Remove the "Question:" and "Context:" parts from the beginning
                     question_input = question_input.replace("Question: ", "")
                     context_input = context_input.replace("Context: ", "")
@@ -206,7 +209,7 @@ if st.session_state.current_conversation:
                     with col1:
                         if st.button("✅ Save", key=f"save_{idx}"):
                             # Combine question and context without the "Question:" and "Context:" labels
-                            new_combined_input = f"{new_question}<br>{new_context}"
+                            new_combined_input = f"{new_question}<br><br>{new_context}"
                             msg["text"] = new_combined_input
                             with st.spinner("Generating response..."):
                                 try:
@@ -255,25 +258,28 @@ if st.session_state.current_conversation:
             st.error("❌ Failed to generate PDF.")
 
     # --- User Prompt ---
-    question_input = st.text_input("Enter your question:")
-    context_input = st.text_area("Enter the context:")
-
-    # Button to submit
-    if st.button("Submit"):
-        if question_input and context_input:
-            combined_input = f"Question: {question_input}<br><br>Context: {context_input}"
-            convo.append({"role": "user", "text": combined_input})
-
-            # Avoid rerunning unnecessarily
-            st.session_state.chat_sessions[st.session_state.current_conversation] = convo
-            st.rerun()
+    with st.form(key="submit_form", clear_on_submit=True):
+        question_input = st.text_input("Enter your question:")
+        context_input = st.text_area("Enter your context:")
+        
+        # Button to submit the form
+        submit_button = st.form_submit_button("Submit")
+        
+        if submit_button:
+            if question_input and context_input:
+                combined_input = f"Question: {question_input}<br><br>Context: {context_input}"
+                convo.append({"role": "user", "text": combined_input})
+
+                # Avoid rerunning unnecessarily
+                st.session_state.chat_sessions[st.session_state.current_conversation] = convo
+                st.rerun()
 
 # Display assistant response after rerun
 if st.session_state.current_conversation and len(st.session_state.chat_sessions[st.session_state.current_conversation]) % 2 == 1:
     convo = st.session_state.chat_sessions[st.session_state.current_conversation]
     last_user_msg = convo[-1]["text"]
 
-    question_input, context_input = last_user_msg.split("<br>")
+    question_input, context_input = last_user_msg.split("<br><br>")
     question_input = question_input.replace("Question: ", "")
     context_input = context_input.replace("Context: ", "")