Update app.py

app.py

@@ -79,6 +79,7 @@ def truncate_document(document, length):
     return document[:length]
 def divide_document(document, max_length):
     return [document[i:i+max_length] for i in range(0, len(document), max_length)]
+
 def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
         data = file.read()
@@ -102,11 +103,6 @@ def get_table_download_link(file_path):
     href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
     return href
 
-    
-
-
-
-
 def CompressXML(xml_text):
     root = ET.fromstring(xml_text)
     for elem in list(root.iter()):
@@ -216,31 +212,31 @@ def main():
     all_files = glob.glob("*.*")
     all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 20]  # exclude files with short names
     all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
-    
+
+    # sidebar of files
     for file in all_files:
-        col1, col3 = st.sidebar.columns([5,1])  # adjust the ratio as needed
+        file_contents=''
+        col1, col2, col3 = st.sidebar.columns([5,1,1])  # adjust the ratio as needed
         with col1:
-            
             st.markdown(get_table_download_link(file), unsafe_allow_html=True)
-            
+        with col2:
             if st.button("🔍", key="read_"+file):  # search emoji button
-                with open(file, 'r') as f:
-                    file_contents = f.read()
-                file_content_area = st.text_area("File Contents:", file_contents, height=100)
-                if st.button('💬 Chat with file content'):
-                    st.write('Reasoning with your inputs...')
-                    response = chat_with_file_contents(user_prompt, file_contents)
-                    st.write('Response:')
-                    st.write(response)
-                    
-                    filename = generate_filename(user_prompt, choice)
-                    create_file(filename, user_prompt, response)
-                    st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-                    
+            with open(file, 'r') as f:
+                file_contents = f.read()
         with col3:
             if st.button("🗑", key="delete_"+file):
                 os.remove(file)
                 st.experimental_rerun()
+        if len(file_contents) > 0:
+            file_content_area = st.text_area("File Contents:", file_contents, height=100)
+            if st.button('💬 Chat with file content'):
+                st.write('Reasoning with your inputs...')
+                response = chat_with_file_contents(file_content_area)
+                st.write('Response:')
+                st.write(response)
+                filename = generate_filename(file_content_area, choice)
+                create_file(filename, file_content_area, response)
+                st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
                 
 if __name__ == "__main__":
     main()