Update app.py

app.py

@@ -147,8 +147,11 @@ for i in range(row_count):
     with col2:
         question = st.text_input(f'Question {i+1}', key=f'question{i}')
     with col3:
+        # Initialize session state for answer if not already done
+        if f'answer{i}' not in st.session_state:
+            st.session_state[f'answer{i}'] = ''
         answer_placeholder = st.empty()  
-        answer_placeholder.text_area(f'Answer {i+1}', key=f'answer{i}') 
+        answer_placeholder.text_area(f'Answer {i+1}', key=f'answer{i}', value=st.session_state[f'answer{i}']) 
     with col4:
         if st.button(f'Submit {i+1}'):
             if openAI_key.strip()=='':
@@ -163,5 +166,6 @@ for i in range(row_count):
                 load_recommender('corpus.pdf')
     
                 answer = generate_answer(question,openAI_key)
-                key = f'answer{i}_{time.time()}'
-                answer_placeholder.text_area(f'Answer {i+1}', key=key, value=answer)
+                # Store the answer in session state
+                st.session_state[f'answer{i}'] = answer
+                answer_placeholder.text_area(f'Answer {i+1}', key=f'answer{i}', value=st.session_state[f'answer{i}'])