Update app.py

app.py

@@ -39,9 +39,8 @@ def get_relevant_context(pdf_text, query, num_contexts=3):
     return relevant_context
 
 # Function to generate a response from GPT-4 chat model
-def generate_response(context, question):
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant expert on GPT-4."},
+def generate_response(context, question, conversation_history):
+    messages = conversation_history + [
         {"role": "user", "content": f"Context: {context}\nQuestion: {question}"}
     ]
     response = openai.ChatCompletion.create(
@@ -50,7 +49,11 @@ def generate_response(context, question):
         max_tokens=1200,
         temperature=0.7,
     )
-    return response['choices'][0]['message']['content'].strip()
+    answer = response['choices'][0]['message']['content'].strip()
+    
+    # Append the new answer to the conversation history
+    conversation_history.append({"role": "assistant", "content": answer})
+    return answer, conversation_history
 
 # Function to handle irrelevant questions
 def is_irrelevant_question(question):
@@ -77,6 +80,10 @@ def main():
             pdf_text = extract_pdf_text(pdf_file)
             st.write("PDF content loaded successfully!")
 
+            # Initialize conversation history (this will persist between interactions)
+            if 'conversation_history' not in st.session_state:
+                st.session_state.conversation_history = []
+            
             # User input: the question they want to ask
             question = st.text_input("Ask your question:")
 
@@ -89,10 +96,21 @@ def main():
                     relevant_context = get_relevant_context(pdf_text, question)
                     
                     # Generate the response from GPT-4 chat model
-                    answer = generate_response(relevant_context, question)
+                    answer, conversation_history = generate_response(relevant_context, question, st.session_state.conversation_history)
+                    
+                    # Update the conversation history in session state
+                    st.session_state.conversation_history = conversation_history
                     
                     # Display the answer
                     st.write(f"Answer: {answer}")
+
+            # End conversation button to reset chat history
+            if st.button("END CONVERSATION"):
+                st.session_state.conversation_history = []  # Reset conversation history
+                st.write("Conversation has been reset. Feel free to ask new questions.")
+            
+        else:
+            st.warning("Please upload a PDF file to proceed.")
     else:
         st.warning("Please enter your OpenAI API Key to use the chatbot.")