Update app.py

app.py

@@ -17,16 +17,16 @@ data_section = st.sidebar.text_area("CSV or Text Data:")
 paste_data = st.sidebar.button("Paste Data")
 num_concurrent_calls = st.sidebar.number_input("Concurrent Calls:", min_value=1, max_value=2000, value=10, step=1)
 generate_all = st.sidebar.button("Generate All")
-reset = st.sidebar.button("Reset")
 add_row = st.sidebar.button("Add row")
+reset = st.sidebar.button("Reset")
+followup_message = st.sidebar.text_area("Edit Message:")
+generate_all_edits = st.sidebar.button("Generate All Edits")
 model = st.sidebar.selectbox("Model:", ["gpt-4-0613", "gpt-4", "gpt-4-32k-0613", "gpt-3.5-turbo", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k"])
 temperature = st.sidebar.slider("Temperature:", 0.0, 1.0, 0.6, step=0.01)
 max_tokens = st.sidebar.number_input("Max Tokens:", min_value=1, max_value=8192, value=1000, step=1)
 top_p = st.sidebar.slider("Top P:", 0.0, 1.0, 1.0, step=0.01)
 system_message = st.sidebar.text_area("System Message:")
-followup_message = st.sidebar.text_area("Edit Message:")
 row_count = st.session_state.get("row_count", 1)
-generate_all_edits = st.sidebar.button("Generate All Edits")
 
 if add_row:
     row_count += 1
@@ -50,7 +50,7 @@ if paste_data:
 if reset:
     row_count = 1
     st.session_state.row_count = row_count
-    for i in range(100): # Assuming a maximum of 100 rows
+    for i in range(1000): 
         st.session_state[f"note{i}"] = ""
         st.session_state[f"message{i}"] = ""
         st.session_state[f"response{i}"] = ""
@@ -95,35 +95,28 @@ class WorkerThread(threading.Thread):
             self.results.put(result)
 
 if generate_all:
-    # Creating a list of messages
     messages = [st.session_state.get(f"message{i}", "") for i in range(row_count)]
 
     jobs = Queue()
     results = Queue()
 
-    # Create workers
     workers = [WorkerThread(jobs, results) for _ in range(num_concurrent_calls)]
 
-    # Start all workers
     for worker in workers:
         worker.start()
 
-    # Put all the jobs into the queue
     for i, message in enumerate(messages):
         jobs.put((i, [
             {"role": "system", "content": system_message},
             {"role": "user", "content": message}
         ]))
 
-    # Put a None for each worker to indicate the end of jobs
     for _ in range(num_concurrent_calls):
         jobs.put(None)
 
-    # Wait for all of the tasks to finish
     for worker in workers:
         worker.join()
 
-    # Collect all the results
     while not results.empty():
         i, response, prompt_tokens, response_tokens, word_count, error_message = results.get()
         if error_message is not None:
@@ -134,20 +127,16 @@ if generate_all:
         st.session_state[f"word_count{i}"] = word_count
 
 if generate_all_edits:
-    # Creating a list of messages
     messages = [st.session_state.get(f"message{i}", "") for i in range(row_count)]
 
     jobs = Queue()
     results = Queue()
 
-    # Create workers
     workers = [WorkerThread(jobs, results) for _ in range(num_concurrent_calls)]
 
-    # Start all workers
     for worker in workers:
         worker.start()
 
-    # Put all the jobs into the queue
     for i, message in enumerate(messages):
         jobs.put((i, [
             {"role": "system", "content": system_message},
@@ -155,15 +144,12 @@ if generate_all_edits:
             {"role": "user", "content": followup_message}
         ]))
 
-    # Put a None for each worker to indicate the end of jobs
     for _ in range(num_concurrent_calls):
         jobs.put(None)
 
-    # Wait for all of the tasks to finish
     for worker in workers:
         worker.join()
 
-    # Collect all the results
     while not results.empty():
         i, response, prompt_tokens, response_tokens, word_count, error_message = results.get()
         if error_message is not None:
@@ -173,8 +159,6 @@ if generate_all_edits:
         st.session_state[f"response_tokens{i}"] = response_tokens
         st.session_state[f"word_count{i}"] = word_count
 
-# ... (rest of the code remains unchanged)
-
 for i in range(row_count):
     if show_notes:
         st.text_input(f"Note {i + 1}:", key=f"note{i}", value=st.session_state.get(f"note{i}", ""))
@@ -198,7 +182,7 @@ for i in range(row_count):
         st.write(f"Tokens: {st.session_state.get(f'prompt_tokens{i}', 0)} / {st.session_state.get(f'response_tokens{i}', 0)} + Words: {st.session_state.get(f'word_count{i}', 0)}")
 
     with col3:
-        st.text_area(f"Follow-up Response {i + 1}:", value=st.session_state.get(f"followup_response{i}", ""))
+        st.text_area(f"Edited Response {i + 1}:", value=st.session_state.get(f"followup_response{i}", ""))
         if st.button(f"Generate Edit {i + 1}"):
             _, followup_response, prompt_tokens, response_tokens, word_count, error_message = generate_response(i, [
                 {"role": "system", "content": system_message},