Update app.py

app.py

@@ -39,8 +39,9 @@ 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, conversation_history):
-    messages = conversation_history + [
+def generate_response(context, question):
+    messages = [
+        {"role": "system", "content": "You are a helpful assistant expert on GPT-4."},
         {"role": "user", "content": f"Context: {context}\nQuestion: {question}"}
     ]
     response = openai.ChatCompletion.create(
@@ -49,11 +50,7 @@ def generate_response(context, question, conversation_history):
         max_tokens=1200,
         temperature=0.7,
     )
-    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
+    return response['choices'][0]['message']['content'].strip()
 
 # Function to handle irrelevant questions
 def is_irrelevant_question(question):
@@ -80,10 +77,6 @@ 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:")
 
@@ -96,23 +89,12 @@ def main():
                     relevant_context = get_relevant_context(pdf_text, question)
                     
                     # Generate the response from GPT-4 chat model
-                    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
+                    answer = generate_response(relevant_context, question)
                     
                     # 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.")
 
 if __name__ == "__main__":
-    main()
+    main()