Update app.py

app.py

@@ -253,18 +253,23 @@ if st.session_state.current_conversation:
     # User Prompt
     if len(convo) % 2 == 1:
         last_user_msg = convo[-1]["text"]
+        question_input, context_input = last_user_msg.split("<br><br>")
+        question_input = question_input.replace("Question: ", "")
+        context_input = context_input.replace("Context: ", "")
+    
+        prompt = create_test_prompt(question_input, context_input, st.session_state.llama_tokenizer)
         with st.spinner("Generating response..."):
             try:
-                assistant_reply = get_wikipedia_summary(last_user_msg)
+                assistant_reply = get_llama_response(prompt)
+                
             except Exception as e:
-                assistant_reply = "⚠️ Failed to generate response"
+                assistant_reply = f"⚠️ Failed to generate response"
         convo.append({"role": "assistant", "text": assistant_reply})
         st.session_state.chat_sessions[st.session_state.current_conversation] = convo
         st.rerun()
 
-    # --- Input Form at Bottom ---
     st.markdown("---")
-    # --- Export PDF ---
+    # Export PDF
     if st.button("📥 Export Conversation as PDF"):
         pdf_bytes = generate_pdf(convo, st.session_state.current_conversation)
         if pdf_bytes: