Update app.py

app.py

@@ -6,7 +6,6 @@ import glob
 import json
 import mistune
 import pytz
-import textwrap
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -64,7 +63,7 @@ def get_table_download_link(file_path):
         data = file.read()
     b64 = base64.b64encode(data.encode()).decode()  
     file_name = os.path.basename(file_path)
-    ext = os.path.splitext(file_name)[1] 
+    ext = os.path.splitext(file_name)[1]  # get the file extension
 
     if ext == '.txt':
         mime_type = 'text/plain'
@@ -73,7 +72,7 @@ def get_table_download_link(file_path):
     elif ext == '.md':
         mime_type = 'text/markdown'
     else:
-        mime_type = 'application/octet-stream' 
+        mime_type = 'application/octet-stream'  # general binary data type
 
     href = f'<a href="data:{mime_type};base64,{b64}" target="_blank" download="{file_name}">{file_name}</a>'
     return href
@@ -107,40 +106,43 @@ def read_file_content(file,max_length):
         return ""
 
 def main():
-    user_prompt = st.text_area("Your question:", '', height=120)
+    prompts = ['']
+    file_content = ""
+    user_prompt = st.text_area("Your question:", '', height=100)
     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...')
-        file_content = ""
+        total_input = "".join(prompts)
+
+        if len(total_input) > 4000:
+            chunks = [total_input[i:i + max_length] for i in range(0, len(total_input), max_length)]
+            responses = []
+            for chunk in chunks:
+                responses.append(chat_with_model([chunk]))
+            response = "\n".join(responses)
+        else:
+            response = chat_with_model(prompts)
+        st.write('Response:')
+        st.write(response)
         
-        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
+        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}")
     
     all_files = glob.glob("*.txt") + glob.glob("*.htm") + glob.glob("*.md")
     for file in all_files:
-        col1, col2 = st.sidebar.columns([4,1]) 
+        col1, col2 = st.sidebar.columns([4,1])  # adjust the ratio as needed
         with col1:
             st.markdown(get_table_download_link(file), unsafe_allow_html=True)
         with col2: