Syncing inputs works.

app/streamlit_meta_prompt.py

@@ -1,3 +1,4 @@
+import pandas as pd
 import streamlit as st
 import json
 import logging
@@ -461,6 +462,8 @@ with st.sidebar:
     aggressive_exploration_input = st.checkbox("Aggressive Exploration", False)
 
 # Initialize session state
+if 'meta_prompt_input_data' not in st.session_state:
+    st.session_state.meta_prompt_input_data = pd.DataFrame(columns=["Input", "Output"])
 if 'initial_system_message' not in st.session_state:
     st.session_state.initial_system_message = ""
 if 'initial_acceptance_criteria' not in st.session_state:
@@ -485,8 +488,17 @@ def copy_acceptance_criteria():
         st.session_state.initial_acceptance_criteria = st.session_state.acceptance_criteria_output
 
 def clear_session_state():
-    for key in st.session_state.keys():
-        del st.session_state[key]
+    st.session_state.meta_prompt_input_data = pd.DataFrame(columns=["Input", "Output"])
+    st.session_state.initial_system_message = ""
+    st.session_state.initial_acceptance_criteria = ""
+    st.session_state.system_message_output = ""
+    st.session_state.output = ""
+    st.session_state.analysis = ""
+    st.session_state.acceptance_criteria_output = ""
+    st.session_state.chat_log = []
+
+def sync_input_data():
+    st.session_state.sample_generator_input_data = data_editor_data.copy()
 
 if active_model_tab == "Simple":
     simple_model_name = simple_model_name_input
@@ -531,10 +543,11 @@ max_output_age = max_output_age_input
 aggressive_exploration = aggressive_exploration_input
 
 data_editor_data = st.data_editor(
-    [{"user_message": "", "expected_output": ""}],
+    st.session_state.meta_prompt_input_data,
+    num_rows="dynamic",
     column_config={
-        "user_message": st.column_config.TextColumn("User Message"),
-        "expected_output": st.column_config.TextColumn("Expected Output"),
+        "Input": st.column_config.TextColumn("Input", width="large"),
+        "Output": st.column_config.TextColumn("Output", width="large"),
     },
     hide_index=True,
     use_container_width=True,
@@ -547,17 +560,19 @@ with col1:
         initial_system_message = st.text_area("Initial System Message", st.session_state.initial_system_message).strip()
         acceptance_criteria = st.text_area("Acceptance Criteria", st.session_state.initial_acceptance_criteria).strip()
     
-    col1_1, col1_2 = st.columns(2)
+    col1_1, col1_2, col1_3 = st.columns(3)
     with col1_1:
         generate_button_clicked = st.button("Generate", type="primary")
     with col1_2:
+        sync_button_clicked = st.button("Sync", on_click=sync_input_data)
+    with col1_3:
         clear_button_clicked = st.button("Clear", on_click=clear_session_state)
 
 with col2:
     if generate_button_clicked:
         try:
-            user_message = data_editor_data[0]["user_message"].strip()
-            expected_output = data_editor_data[0]["expected_output"].strip()
+            user_message = data_editor_data["Input"][0].strip()
+            expected_output = data_editor_data["Output"][0].strip()
 
             if active_model_tab == "Simple":
                 system_message, output, analysis, acceptance_criteria, chat_log = process_message_with_single_llm(

app/streamlit_sample_generator.py

@@ -144,8 +144,8 @@ def example_selected():
 
 
 # Session State
-if 'input_data' not in st.session_state:
-    st.session_state.input_data = pd.DataFrame(columns=["Input", "Output"])
+if 'sample_generator_input_data' not in st.session_state:
+    st.session_state.sample_generator_input_data = pd.DataFrame(columns=["Input", "Output"])
 
 if 'description_output_text' not in st.session_state:
     st.session_state.description_output_text = ''
@@ -245,7 +245,7 @@ def import_input_data_from_json():
             data = st.session_state.input_file.getvalue()
             data = json.loads(data)
             data = [{k.capitalize(): v for k, v in d.items()} for d in data]
-            st.session_state.input_data = pd.DataFrame(data)
+            st.session_state.sample_generator_input_data = pd.DataFrame(data)
     except Exception as e:
         st.warning(f"Failed to import JSON: {str(e)}")
 
@@ -277,13 +277,28 @@ def add_new_suggestion():
         st.session_state.suggestions.append(st.session_state.new_suggestion)
         st.session_state.new_suggestion = ""  # Clear the input field
 
+def sync_input_data():
+    st.session_state.meta_prompt_input_data = input_data.copy()
+    st.session_state.sample_generator_input_data = input_data.copy()
+
+def clear_session_state():
+    st.session_state.sample_generator_input_data = pd.DataFrame(columns=["Input", "Output"])
+    st.session_state.description_output_text = ''
+    st.session_state.suggestions = []
+    st.session_state.input_analysis_output_text = ''
+    st.session_state.example_briefs_output_text = ''
+    st.session_state.examples_from_briefs_dataframe = pd.DataFrame(columns=["Input", "Output"])
+    st.session_state.examples_directly_dataframe = pd.DataFrame(columns=["Input", "Output"])
+    st.session_state.examples_dataframe = pd.DataFrame(columns=["Input", "Output"])
+    st.session_state.selected_example = None
+
 # Streamlit UI
 st.title("LLM Task Example Generator")
 st.markdown("Enter input-output pairs in the table below to generate a task description, analysis, and additional examples.")
 
 # Input column
 input_data = st.data_editor(
-    st.session_state.input_data,
+    st.session_state.sample_generator_input_data,
     num_rows="dynamic",
     use_container_width=True,
     column_config={
@@ -315,8 +330,16 @@ with st.expander("Model Settings"):
     temperature = st.slider("Temperature", 0.0, 1.0, 1.0, 0.1)
     generating_batch_size = st.slider("Generating Batch Size", 1, 10, 3, 1)
 
-submit_button = st.button(
-    "Generate", type="primary", on_click=generate_examples_dataframe)
+col1, col2, col3 = st.columns(3)
+with col1:
+    submit_button = st.button(
+        "Generate", type="primary", on_click=generate_examples_dataframe)
+with col2:
+    sync_button = st.button(
+        "Sync", on_click=sync_input_data)
+with col3:
+    clear_button = st.button(
+        "Clear", on_click=clear_session_state)
 
 with st.expander("Description and Analysis"):
     generate_description_button = st.button(
@@ -365,8 +388,8 @@ examples_output = st.dataframe(st.session_state.examples_dataframe, use_containe
 
 def append_selected_to_input_data():
     if st.session_state.selected_example is not None:
-        st.session_state.input_data = pd.concat(
-            [st.session_state.input_data, st.session_state.selected_example], ignore_index=True)
+        st.session_state.sample_generator_input_data = pd.concat(
+            [st.session_state.sample_generator_input_data, st.session_state.selected_example], ignore_index=True)
         st.session_state.selected_example = None
 
 def show_sidebar():