Update app.py

app.py

@@ -3,7 +3,6 @@ import openai
 import os
 import base64
 import glob
-import json
 import mistune
 import pytz
 import math
@@ -18,30 +17,6 @@ st.set_page_config(
     page_title="GPT Streamlit Document Reasoner",
     layout="wide")
 
-# Custom CSS to increase scrollbar width
-st.markdown("""
-    <style>
-    ::-webkit-scrollbar {
-      width: 20px;
-    }
-
-    /* Track */
-    ::-webkit-scrollbar-track {
-      background: #f1f1f1;
-    }
-
-    /* Handle */
-    ::-webkit-scrollbar-thumb {
-      background: #888;
-    }
-
-    /* Handle on hover */
-    ::-webkit-scrollbar-thumb:hover {
-      background: #555;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
 menu = ["txt", "htm", "md", "py"]
 choice = st.sidebar.selectbox("Output file type:", menu)
 choicePrefix = "Output file type is "
@@ -54,6 +29,9 @@ elif choice == "md":
 elif choice == "py":
      st.sidebar.write(choicePrefix + "Python Code.")
 
+# Create 3 columns with column 2 being twice as large
+col1, col2, col3 = st.columns([1, 2, 1])
+
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -71,6 +49,9 @@ def create_file(filename, prompt, response):
         with open(filename, 'w') as file:
             file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
 
+def divide_document(document, max_length):
+    return [document[i:i+max_length] for i in range(0, len(document), max_length)]
+
 def chat_with_model(prompt, document_section):
     model = "gpt-3.5-turbo"
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
@@ -106,7 +87,7 @@ def read_file_content(file):
     elif file.type == "application/xml" or file.type == "text/xml":
         tree = ET.parse(file)
         root = tree.getroot()
-        xml = ET.tostring(root, encoding='unicode')
+        xml = CompressXML(ET.tostring(root, encoding='unicode'))
         return xml
     elif file.type == "text/markdown" or file.type == "text/md":
         md = mistune.create_markdown()
@@ -118,43 +99,37 @@ def read_file_content(file):
         return ""
 
 def main():
-    col1, col2 = st.columns([1, 2])
-
-    with col1:
-        user_prompt = st.text_area("Your question:", '', height=150)  # Increased height for question text box
-        uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
+    user_prompt = col1.text_area("Your question:", '', height=150)
+    uploaded_file = col1.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
+    max_length = 4000
 
     document_sections = deque()
     document_responses = {}
 
     if uploaded_file is not None:
         file_content = read_file_content(uploaded_file)
-        document_sections.append(file_content)
-
-    with col2:
-        if st.button('💬 Chat'):
-            st.write('Thinking and Reasoning with your inputs...')
-            response = chat_with_model(user_prompt, ''.join(list(document_sections)))
-            response_area = st.text_area('Response:', value=response, height=400)
-            filename = generate_filename(user_prompt, choice)
-            create_file(filename, user_prompt, response)
-            st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+        document_sections.extend(divide_document(file_content, max_length))
 
     if len(document_sections) > 0:
-        for i, section in reversed(list(enumerate(list(document_sections)))):
+        for i, section in enumerate(list(document_sections)):
             if i in document_responses:
-                st.markdown(f"**Section {i+1} Content:**\n{section}")
-                response_area = st.text_area(f"Section {i+1} Response:", value=document_responses[i], height=400)
+                col2.text(f"Section {i+1} Response")
+                col2.text_area('', document_responses[i], height=200)
             else:
-                if st.button(f"Chat about Section {i+1}"):
-                    st.write('Thinking and Reasoning with your inputs...')
+                if col3.button(f"Chat about Section {i+1}"):
                     response = chat_with_model(user_prompt, section)
                     document_responses[i] = response
-                    response_area = st.text_area(f"Section {i+1} Response:", value=response, height=400)
                     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 col3.button('💬 Chat'):
+        response = chat_with_model(user_prompt, ''.join(list(document_sections)))
+        document_responses['aggregate'] = 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)
+    
     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