Update app.py

app.py

@@ -6,6 +6,7 @@ import glob
 import json
 import mistune
 import pytz
+import textwrap
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -58,21 +59,12 @@ def create_file(filename, prompt, response):
         with open(filename, 'w') as file:
             file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
 
-def get_table_download_link_old(file_path):
-    with open(file_path, 'r') as file:
-        data = file.read()
-    b64 = base64.b64encode(data.encode()).decode()  
-    href = f'<a href="data:file/htm;base64,{b64}" target="_blank" download="{os.path.basename(file_path)}">{os.path.basename(file_path)}</a>'
-    return href
-
 def get_table_download_link(file_path):
-    import os
-    import base64
     with open(file_path, 'r') as file:
         data = file.read()
     b64 = base64.b64encode(data.encode()).decode()  
     file_name = os.path.basename(file_path)
-    ext = os.path.splitext(file_name)[1]  # get the file extension
+    ext = os.path.splitext(file_name)[1] 
 
     if ext == '.txt':
         mime_type = 'text/plain'
@@ -81,7 +73,7 @@ def get_table_download_link(file_path):
     elif ext == '.md':
         mime_type = 'text/markdown'
     else:
-        mime_type = 'application/octet-stream'  # general binary data type
+        mime_type = 'application/octet-stream' 
 
     href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
     return href
@@ -115,34 +107,40 @@ def read_file_content(file,max_length):
         return ""
 
 def main():
-    prompts = ['']
-    file_content = ""
     user_prompt = st.text_area("Your question:", '', height=120)
     uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
 
-    if user_prompt:
-        prompts.append(user_prompt)
-
-    if uploaded_file is not None:
-        file_content = read_file_content(uploaded_file, max_length)
-        prompts.append(file_content)
-
     if st.button('💬 Chat'):
         st.write('Thinking and Reasoning with your inputs...')
-        response = chat_with_model(prompts)
-        st.write('Response:')
-        st.write(response)
+        file_content = ""
         
-        filename = generate_filename(user_prompt, choice)
-        create_file(filename, user_prompt, response)
-        st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-    
-    if len(file_content) > 0:
-        st.markdown(f"**File Content Added:**\n{file_content}")
+        if user_prompt:
+            prompts = textwrap.wrap(user_prompt, max_length)
+            for prompt in prompts:
+                response = chat_with_model([prompt])
+                st.write('Response:')
+                st.write(response)
+                filename = generate_filename(prompt, choice)
+                create_file(filename, prompt, response)
+                st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+
+        if uploaded_file is not None:
+            file_content = read_file_content(uploaded_file, max_length)
+            document_parts = textwrap.wrap(file_content, max_length)
+            for part in document_parts:
+                response = chat_with_model([part])
+                st.write('Response:')
+                st.write(response)
+                filename = generate_filename(part, choice)
+                create_file(filename, part, response)
+                st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+        
+        if len(file_content) > 0:
+            st.text_area("File Content:", file_content, height=400)  # Display file content in a scrollable text box
     
     all_files = glob.glob("*.txt") + glob.glob("*.htm") + glob.glob("*.md")
     for file in all_files:
-        col1, col2 = st.sidebar.columns([4,1])  # adjust the ratio as needed
+        col1, col2 = st.sidebar.columns([4,1]) 
         with col1:
             st.markdown(get_table_download_link(file), unsafe_allow_html=True)
         with col2:
@@ -151,4 +149,4 @@ def main():
                 st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
+    main()