Update app.py

app.py

@@ -59,10 +59,22 @@ def generate_summary_with_groq(query, retrieved_text):
 def main():
     st.title("Movie Analysis: RAG-based Application")
     
+    # Initialize session state variables
+    if 'questions' not in st.session_state:
+        st.session_state.questions = []
+        
+    if 'summaries' not in st.session_state:
+        st.session_state.summaries = []
+
     # User input
     user_query = st.text_input("Ask a question about movies:")
     
     if user_query:
+        # Check if user wants to exit
+        if user_query.lower() in ['exit', 'no', 'quit']:
+            st.write("Exiting the application. Goodbye!")
+            return
+        
         # Retrieve relevant movie summaries
         retrieved_movies = retrieve_similar_movies(user_query, movies_df, tfidf_matrix)
         retrieved_summaries = " ".join(retrieved_movies['summary'].values)
@@ -70,6 +82,10 @@ def main():
         # Generate a summary response based on retrieved movies
         generated_summary = generate_summary_with_groq(user_query, retrieved_summaries)
         
+        # Store the question and summary in session state
+        st.session_state.questions.append(user_query)
+        st.session_state.summaries.append(generated_summary)
+
         # Display the generated summary
         st.subheader("Generated Summary:")
         st.write(generated_summary)
@@ -77,5 +93,12 @@ def main():
         # Provide an option for the user to ask another question
         st.text("You can ask another question.")
 
+        # Display the previous questions and summaries
+        if st.session_state.questions:
+            st.write("### Previous Questions and Summaries:")
+            for question, summary in zip(st.session_state.questions, st.session_state.summaries):
+                st.write(f"- **Q:** {question}")
+                st.write(f"  **A:** {summary}")
+
 if __name__ == "__main__":
     main()