Update app.py

app.py

@@ -249,6 +249,21 @@ if st.session_state.current_conversation:
 
     st.markdown('</div>', unsafe_allow_html=True)
 
+    # --- User Prompt ---
+    # --- Generate Assistant Response ---
+    if len(convo) % 2 == 1:
+        last_user_msg = convo[-1]["text"]
+        with st.spinner("Generating response..."):
+            try:
+                assistant_reply = get_wikipedia_summary(last_user_msg)
+            except Exception as e:
+                assistant_reply = "⚠️ 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 ---
     if st.button("📥 Export Conversation as PDF"):
         pdf_bytes = generate_pdf(convo, st.session_state.current_conversation)
@@ -256,42 +271,14 @@ if st.session_state.current_conversation:
             st.download_button("Download PDF", pdf_bytes, file_name="AskWikipedia_Conversation.pdf", mime="application/pdf")
         else:
             st.error("❌ Failed to generate PDF.")
-
-    # --- User Prompt ---
+            
     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}"
+                combined_input = f"Question: {question_input}<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><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_llama_response(prompt)
-            
-        except Exception as e:
-            assistant_reply = f"⚠️ Failed to generate response"
-
-    # Now, append the response after the spinner
-    convo.append({"role": "assistant", "text": assistant_reply})
-    st.session_state.chat_sessions[st.session_state.current_conversation] = convo
-    st.rerun()
+                st.rerun()