Update app.py

app.py

@@ -30,8 +30,6 @@ elif choice == "md":
 elif choice == "py":
      st.sidebar.write(choicePrefix + "Python Code.")
 
-max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=2000, step=1000)
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -49,12 +47,6 @@ def create_file(filename, prompt, response):
         with open(filename, 'w') as file:
             file.write(f"# Prompt:\n{prompt}\n# Response:\n{response}")
 
-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 chat_with_model(prompt, document_section):
     model = "gpt-3.5-turbo"
     conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
@@ -63,7 +55,6 @@ def chat_with_model(prompt, document_section):
     response = openai.ChatCompletion.create(model=model, messages=conversation)
     return response['choices'][0]['message']['content']
 
-
 def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
         data = file.read()
@@ -81,15 +72,7 @@ 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()):
-        if isinstance(elem.tag, str) and 'Comment' in elem.tag:
-            elem.parent.remove(elem)
-    return ET.tostring(root, encoding='unicode', method="xml")
-    
-def read_file_content(file,max_length):
+def read_file_content(file):
     if file.type == "application/json":
         content = json.load(file)
         return str(content)
@@ -99,7 +82,7 @@ def read_file_content(file,max_length):
     elif file.type == "application/xml" or file.type == "text/xml":
         tree = ET.parse(file)
         root = tree.getroot()
-        xml = CompressXML(ET.tostring(root, encoding='unicode'))
+        xml = ET.tostring(root, encoding='unicode')
         return xml
     elif file.type == "text/markdown" or file.type == "text/md":
         md = mistune.create_markdown()
@@ -111,44 +94,49 @@ def read_file_content(file,max_length):
         return ""
 
 def main():
-    user_prompt = st.text_area("Your question:", '', height=120)
-    uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
-    max_length = 4000
+    col1, col2, col3 = st.columns([1, 1, 1])
+
+    with col1:
+        user_prompt = st.text_area("Your question:", '', height=120)
+        uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "html", "htm", "md", "txt"])
 
     document_sections = deque()
     document_responses = {}
 
     if uploaded_file is not None:
-        file_content = read_file_content(uploaded_file, max_length)
-        document_sections.extend(divide_document(file_content, max_length))
+        file_content = read_file_content(uploaded_file)
+        document_sections.append(file_content)
 
     if len(document_sections) > 0:
-        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]}")
-            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)
-                    st.write('Response:')
-                    st.write(response)
-                    document_responses[i] = 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)
-
+        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]}")
+                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)
 
     if st.button('💬 Chat'):
         st.write('Thinking and Reasoning with your inputs...')
         response = chat_with_model(user_prompt, ''.join(list(document_sections)))
         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)