Update app.py

app.py

@@ -94,7 +94,7 @@ def read_file_content(file):
         return ""
 
 def main():
-    col1, col2, col3 = st.columns([1, 1, 1])
+    col1, col2 = st.columns([1, 2])
 
     with col1:
         user_prompt = st.text_area("Your question:", '', height=120)
@@ -109,27 +109,21 @@ def main():
 
     if len(document_sections) > 0:
         with col2:
-            st.markdown("**Sections of the uploaded file:**")
-            for i, section in enumerate(list(document_sections)):
-                st.markdown(f"**Section {i+1}**\n{section}")
-
             st.markdown("**Chat with the model:**")
             for i, section in enumerate(list(document_sections)):
                 if i in document_responses:
-                    st.markdown(f"**Section {i+1}**\n{document_responses[i]}")
+                    st.markdown(f"**Section {i+1} Response:**\n{document_responses[i]}")
+                    st.markdown(f"**Section {i+1} Content:**\n{section}")
                 else:
                     if st.button(f"Chat about Section {i+1}"):
                         st.write('Thinking and Reasoning with your inputs...')
                         response = chat_with_model(user_prompt, section)
                         document_responses[i] = response
-
-        with col3:
-            st.markdown("**Responses from the model:**")
-            for i, response in enumerate(document_responses.values()):
-                st.markdown(f"**Response to Section {i+1}**\n{response}")
-                filename = generate_filename(f"{user_prompt}_section_{i+1}", choice)
-                create_file(filename, user_prompt, response)
-                st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+                        st.markdown(f"**Section {i+1} Response:**\n{response}")
+                        st.markdown(f"**Section {i+1} Content:**\n{section}")
+                        filename = generate_filename(f"{user_prompt}_section_{i+1}", choice)
+                        create_file(filename, user_prompt, response)
+                        st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
     if st.button('💬 Chat'):
         st.write('Thinking and Reasoning with your inputs...')
@@ -152,4 +146,4 @@ def main():
                 st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
+    main()