Update app.py

app.py

@@ -6,11 +6,12 @@ import glob
 import json
 import mistune
 import pytz
-
+import math
 from datetime import datetime
 from openai import ChatCompletion
 from xml.etree import ElementTree as ET
 from bs4 import BeautifulSoup
+from collections import deque
 
 openai.api_key = os.getenv('OPENAI_KEY')
 st.set_page_config(
@@ -31,16 +32,6 @@ elif choice == "py":
 
 max_length = st.sidebar.slider("Max document length", min_value=1000, max_value=32000, value=2000, step=1000)
 
-def truncate_document(document, length):
-    return document[:length]
-
-def chat_with_model(prompts):
-    model = "gpt-3.5-turbo"
-    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
-    conversation.extend([{'role': 'user', 'content': prompt} for prompt in prompts])
-    response = openai.ChatCompletion.create(model=model, messages=conversation)
-    return response['choices'][0]['message']['content']
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -58,12 +49,20 @@ 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 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.'}]
+    conversation.append({'role': 'user', 'content': prompt})
+    conversation.append({'role': 'assistant', 'content': 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:
@@ -82,6 +81,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()):
@@ -111,31 +111,44 @@ 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"])
+    max_length = 4000
 
-    if user_prompt:
-        prompts.append(user_prompt)
+    document_sections = deque()
+    document_responses = {}
 
     if uploaded_file is not None:
         file_content = read_file_content(uploaded_file, max_length)
-        prompts.append(file_content)
+        document_sections.extend(divide_document(file_content, max_length))
+
+    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
 
     if st.button('💬 Chat'):
         st.write('Thinking and Reasoning with your inputs...')
-        response = chat_with_model(prompts)
+        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)
-    
-    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])  # adjust the ratio as needed
@@ -147,4 +160,4 @@ def main():
                 st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
+    main()