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-Chat-v0.6-rag-finetunedv1.0")
-    model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Chat-v0.6-rag-finetunedv1.0")
+    tokenizer = AutoTokenizer.from_pretrained("istiak101/TinyLlama-1.1B-Finetuned")
+    model = AutoModelForCausalLM.from_pretrained("istiak101/TinyLlama-1.1B-Finetuned")
     return model, tokenizer
 
 def create_test_prompt(question, context, tokenizer):
@@ -79,6 +79,14 @@ def generate_pdf(convo, topic):
         return None
     return result
 
+def submit1():
+    st.session_state.question_input = st.session_state.qi
+    st.session_state.qi = ""
+
+def submit2():
+    st.session_state.context_input = st.session_state.ci
+    st.session_state.ci = ""
+
 # --- Session Init ---
 if "chat_sessions" not in st.session_state:
     st.session_state.chat_sessions = {}
@@ -255,17 +263,18 @@ 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:")
+    # question_input = st.text_input("Enter your question:")
+    # context_input = st.text_area("Enter the context:")
+
+    st.text_input("Enter your question:", key="qi", on_change=submit1)
+    st.text_area("Enter the context:", key="ci", on_change=submit2)
 
     # Button to submit
     if st.button("Submit"):
         if question_input and context_input:
-            combined_input = f"Question: {question_input}<br>Context: {context_input}"
+            combined_input = f"Question: {question_input}<br><br>Context: {context_input}"
             convo.append({"role": "user", "text": combined_input})
 
-            st.session_state.question_input = ""
-            st.session_state.context_input = ""
             # Avoid rerunning unnecessarily
             st.session_state.chat_sessions[st.session_state.current_conversation] = convo
             st.rerun()