Update app.py

app.py

@@ -73,14 +73,11 @@ if submit_button and query:
     if uploaded_files:
         pdf_text = extract_text_from_pdfs(uploaded_files)
         
-        # Prepare the input prompt
+        # Prepare the input prompt without unnecessary headers
         prompt = f"""
-        ### Instruction and Input:
         Based on the following context/document:
         {pdf_text}
         Please answer the question: {query}
-
-        ### Response:
         """
 
         # Encode the input text
@@ -97,15 +94,16 @@ if submit_button and query:
             no_repeat_ngram_size=5,
         )
 
-        # Decode the response
+        # Decode and clean the response
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        response_clean = response.replace("### Instruction and Input:", "").replace("### Response:", "").strip()
 
         # Update chat history
-        st.session_state.chat_history.append((query, response))
+        st.session_state.chat_history.append((query, response_clean))
 
 # Display chat history
 if st.session_state.chat_history:
-    for i, (q, a) in enumerate(st.session_state.chat_history):
-        st.markdown(f"**Question {i + 1}:** {q}")
+    for q, a in st.session_state.chat_history:
+        st.markdown(f"**Question:** {q}")
         st.markdown(f"**Answer:** {a}")
         st.write("---")