updated gui

src/Start_Page.py

@@ -9,6 +9,41 @@ from qstn.prompt_builder import LLMPrompt
 st.set_page_config(layout="wide")
 st.title("Questionnaire")
 
+# Demo section showing expected CSV format
+with st.expander("📋 View Example CSV Format", expanded=False):
+    demo_col1, demo_col2 = st.columns(2)
+    
+    with demo_col1:
+        st.subheader("Questionnaire CSV Format")
+        st.write("**Required columns:** `questionnaire_item_id`, `question_content`, `question_stem`")
+        # Create example dataframe
+        example_questionnaire = pd.DataFrame({
+            'questionnaire_item_id': [1, 2, 3],
+            'question_content': ['Coffee', 'Pizza', 'Ice cream'],
+            'question_stem': [
+                'How do you feel about?',
+                'How do you feel about?',
+                'How do you feel about?'
+            ]
+        })
+        st.write(example_questionnaire)
+    
+    with demo_col2:
+        st.subheader("Population CSV Format")
+        st.write("**Required columns:** `questionnaire_name`, `system_prompt`, `questionnaire_instruction`")
+        # Create example dataframe
+        example_population = pd.DataFrame({
+            'questionnaire_name': ['Student', 'Teacher'],
+            'system_prompt': ['You are a student.', 'You are a teacher.'],
+            'questionnaire_instruction': [
+                'Please answer the following questions.',
+                'Please answer the following questions.'
+            ]
+        })
+        st.write(example_population)
+
+st.divider()
+
 col1, col2 = st.columns(2)
 
 # def save_dataframe(uploaded_file):
@@ -22,39 +57,60 @@ col1, col2 = st.columns(2)
 
 #     return df
 
+# Default example dataframes
+default_questionnaire = pd.DataFrame({
+    'questionnaire_item_id': [1, 2, 3],
+    'question_content': ['Coffee', 'Pizza', 'Ice cream'],
+    'question_stem': [
+        'How do you feel about?',
+        'How do you feel about?',
+        'How do you feel about?'
+    ]
+})
+
+default_population = pd.DataFrame({
+    'questionnaire_name': ['Student', 'Teacher'],
+    'system_prompt': ['You are a student.', 'You are a teacher.'],
+    'questionnaire_instruction': [
+        'Please answer the following questions.',
+        'Please answer the following questions.'
+    ]
+})
+
 df_population = None
 df_questionnaire = None
+using_defaults = False
 
 with col1:
     uploaded_questionnaire = st.file_uploader("Select a questionnaire to start with")
     if uploaded_questionnaire is not None:
-        # bytes_data = uploaded_questionnaire.getvalue()
-        # st.write(bytes_data)
-
         df_questionnaire = pd.read_csv(uploaded_questionnaire)
-        #dataframe = save_dataframe(uploaded_questionnaire)
-
+        st.write(df_questionnaire)
+    else:
+        # Use default if no file uploaded
+        df_questionnaire = default_questionnaire
+        using_defaults = True
+        st.info("ℹ️ Using example questionnaire. Upload a file to use your own data.")
         st.write(df_questionnaire)
 
 with col2:
     uploaded_population = st.file_uploader("Select a population to start with")
     if uploaded_population is not None:
-        # bytes_data = uploaded_population.getvalue()
-        # st.write(bytes_data)
-
         df_population = pd.read_csv(uploaded_population)
-        #dataframe = save_dataframe(uploaded_population)
-
+        st.write(df_population)
+    else:
+        # Use default if no file uploaded
+        df_population = default_population
+        using_defaults = True
+        st.info("ℹ️ Using example population. Upload a file to use your own data.")
         st.write(df_population)
 
-disabled = True
-
-if df_population is not None and df_questionnaire is not None:
-    disabled = False
+# Button is always enabled now (using defaults if no files uploaded)
+disabled = False
         
 st.divider()
 
 if st.button("Confirm and Prepare Questionnaire", type="primary", disabled=disabled, use_container_width=True):
     questionnaires: list[LLMPrompt] = SurveyCreator.from_dataframe(df_population, df_questionnaire)
     st.session_state.questionnaires = questionnaires
-    st.switch_page("pages/01_Prompt_Configuration.py")
+    st.switch_page("pages/01_Option_Prompt.py")

src/pages/{02_Option_Prompt.py → 01_Option_Prompt.py}

@@ -145,9 +145,9 @@ with st.container(border=True):
     # The submit button for the form
     submitted = st.button("Confirm and Generate Options", disabled=disabled, type="primary", use_container_width=True)
 
-    if st.button("Remove all options", use_container_width=True, icon="❌"):
+    if st.button("Skip", use_container_width=True, icon="❌"):
         st.session_state.survey_options = None
-        st.switch_page("pages/03_Inference_Setting.py")
+        st.switch_page("pages/02_Prompt_Configuration.py")
 
 # --- Processing and Output ---
 if submitted:
@@ -192,6 +192,6 @@ if submitted:
         )
 
         st.session_state.survey_options = survey_options
-        st.switch_page("pages/03_Inference_Setting.py")
+        st.switch_page("pages/02_Prompt_Configuration.py")
 
 

src/pages/{01_Prompt_Configuration.py → 02_Prompt_Configuration.py}

@@ -23,7 +23,7 @@ st.write(
     "This interface allows you configure how the questions are prompted to the LLM and the overall prompt structure. "
     "These options are applied to every questionnaire in your survey."
 )
-st.page_link("pages/02_Option_Prompt.py", label="Click here to adjust the answer options.")
+st.page_link("pages/01_Option_Prompt.py", label="Click here to adjust the answer options.")
 st.divider()
 
 @st.cache_data
@@ -104,6 +104,117 @@ if "questionnaires" in st.session_state and st.session_state.questionnaires is n
 
     with col1:
         st.subheader("⚙️ Configuration")
+        
+        # Global settings checkboxes at the top
+        change_all_system = state.create(
+            st.checkbox,
+            key=change_all_system_prompts_checkbox,
+            label="On update: change all System Prompts",
+            help="If this is ticked, all system prompts will be changed to this.",
+            initial_value=True
+        )
+        
+        change_all_questionnaire = state.create(
+            st.checkbox,
+            key=change_all_prompts_checkbox,
+            label="On update: change all questionnaire instructions",
+            help="If this is ticked, all questionnaire instructions will be changed to this.",
+            initial_value=True
+        )
+        
+        st.divider()
+
+        # System prompt and main prompt section (from Basic Prompt Settings)
+        system_prompt_key = f"{system_prompt_field}{current_questionnaire_id}"
+        prompt_key = f"{prompt_field}{current_questionnaire_id}"
+        
+        # Handle placeholder replacement for both textboxes before widgets are created
+        if "unified_placeholder_to_replace" in st.session_state and st.session_state.unified_placeholder_to_replace:
+            current_system_text = st.session_state.get(system_prompt_key, questionnaire.system_prompt)
+            current_prompt_text = st.session_state.get(prompt_key, questionnaire.prompt)
+            placeholder_shortcut = st.session_state.unified_placeholder_to_replace["shortcut"]
+            placeholder_value = st.session_state.unified_placeholder_to_replace["value"]
+            target_textbox = st.session_state.unified_placeholder_to_replace.get("target", "prompt")  # "system" or "prompt"
+            
+            # Check if placeholder already exists in EITHER textbox
+            if placeholder_value in current_system_text or placeholder_value in current_prompt_text:
+                st.session_state.unified_placeholder_warning = f"⚠️ The placeholder `{placeholder_value}` already exists in one of the textboxes. Please remove it first if you want to insert it again."
+            else:
+                # Check for shortcut in both textboxes - replace where found
+                if placeholder_shortcut in current_system_text:
+                    st.session_state[system_prompt_key] = current_system_text.replace(placeholder_shortcut, placeholder_value)
+                elif placeholder_shortcut in current_prompt_text:
+                    st.session_state[prompt_key] = current_prompt_text.replace(placeholder_shortcut, placeholder_value)
+                else:
+                    # No shortcut found, add to the target textbox (default based on which was clicked)
+                    if target_textbox == "system":
+                        st.session_state[system_prompt_key] = current_system_text + f" {placeholder_value} "
+                    else:
+                        st.session_state[prompt_key] = current_prompt_text + f" {placeholder_value} "
+            
+            st.session_state.unified_placeholder_to_replace = None
+            st.rerun()
+        
+        # Display warnings if they exist
+        if "unified_placeholder_warning" in st.session_state and st.session_state.unified_placeholder_warning:
+            st.warning(st.session_state.unified_placeholder_warning)
+            st.session_state.unified_placeholder_warning = None  # Clear after displaying
+        
+        new_system_prompt = st.text_area(
+            label=system_prompt_field,
+            key=system_prompt_key,
+            value=questionnaire.system_prompt,
+            help="The system prompt the model is prompted with."
+        )
+
+        # Display warning for main prompt if it exists
+        if "main_prompt_warning" in st.session_state and st.session_state.main_prompt_warning:
+            st.warning(st.session_state.main_prompt_warning)
+            st.session_state.main_prompt_warning = None  # Clear after displaying
+        
+        new_prompt = st.text_area(
+            label=prompt_field,
+            key=prompt_key,
+            value=questionnaire.prompt,
+            help="Instructions that are given to the model before the questions."
+        )
+
+        # Unified placeholder insertion buttons (work for both system prompt and main prompt)
+        st.write("**Insert Placeholder:**")
+        
+        # Check if options were configured (needed for both main prompt and question stem sections)
+        options_configured = "survey_options" in st.session_state and st.session_state.survey_options is not None
+        
+        unified_placeholders = [
+            (placeholder.PROMPT_QUESTIONS, "-P", "P", "Prompt Questions"),
+            (placeholder.PROMPT_OPTIONS, "-O", "O", "Prompt Options"),
+            (placeholder.PROMPT_AUTOMATIC_OUTPUT_INSTRUCTIONS, "-A", "A", "Automatic Output"),
+            (placeholder.JSON_TEMPLATE, "-J", "J", "JSON Template"),
+        ]
+        
+        shortcuts_list = ", ".join([f"`{shortcut}`" for _, shortcut, _, _ in unified_placeholders])
+        st.caption(f"💡 Tip: Type shortcuts {shortcuts_list} in either the system prompt or main prompt, then click the button to replace them. The placeholder will be inserted where the shortcut is found, or in the main prompt if not found.")
+        
+        cols_unified = st.columns(len(unified_placeholders))
+        for i, (placeholder_value, shortcut, char_label, description) in enumerate(unified_placeholders):
+            button_key = f"btn_unified_placeholder_{char_label}"
+            
+            # Disable "Prompt Options" button if options weren't configured
+            is_options_button = placeholder_value == placeholder.PROMPT_OPTIONS
+            button_disabled = is_options_button and not options_configured
+            button_help = f"Replaces '{shortcut}' with {placeholder_value} in either textbox"
+            if button_disabled:
+                button_help = "⚠️ You need to configure answer options first. Go back to the Options page to set them up."
+            
+            if cols_unified[i].button(description, key=button_key, use_container_width=True, disabled=button_disabled, help=button_help):
+                st.session_state.unified_placeholder_to_replace = {
+                    "shortcut": shortcut,
+                    "value": placeholder_value,
+                    "target": "prompt"  # Default to main prompt if no shortcut found
+                }
+                st.rerun()
+
+        st.divider()
 
         for field_id in field_ids:
             input_key = f"input_{field_id}"
@@ -120,17 +231,26 @@ if "questionnaires" in st.session_state and st.session_state.questionnaires is n
             placeholder_shortcut = st.session_state.placeholder_to_replace["shortcut"]
             placeholder_value = st.session_state.placeholder_to_replace["value"]
             
-            # Replace all occurrences of the shortcut (e.g., -Q) with the placeholder
-            if placeholder_shortcut in current_text:
-                st.session_state[input_key] = current_text.replace(placeholder_shortcut, placeholder_value)
+            # Check if placeholder already exists in the text
+            if placeholder_value in current_text:
+                st.session_state.question_stem_warning = f"⚠️ The placeholder `{placeholder_value}` already exists in the text. Please remove it first if you want to insert it again."
             else:
-                # Shortcut not found, append at the end
-                st.session_state[input_key] = current_text + f" {placeholder_value} "
+                # Replace all occurrences of the shortcut (e.g., -Q) with the placeholder
+                if placeholder_shortcut in current_text:
+                    st.session_state[input_key] = current_text.replace(placeholder_shortcut, placeholder_value)
+                else:
+                    # Shortcut not found, append at the end
+                    st.session_state[input_key] = current_text + f" {placeholder_value} "
             
             st.session_state.placeholder_to_replace = None
             st.rerun()
 
         # --- Input Widgets (No Form) ---
+        # Display warning for question stem if it exists
+        if "question_stem_warning" in st.session_state and st.session_state.question_stem_warning:
+            st.warning(st.session_state.question_stem_warning)
+            st.session_state.question_stem_warning = None  # Clear after displaying
+        
         question_stem_input = st.text_area(
             question_stem_field,
             key=f"input_{question_stem_field}",
@@ -160,7 +280,14 @@ if "questionnaires" in st.session_state and st.session_state.questionnaires is n
             button_label = description  # Use the actual placeholder name
             button_key = f"btn_placeholder_{char_label}"
             
-            if cols[i].button(button_label, key=button_key, use_container_width=True, help=f"Replaces '{shortcut}' with {placeholder_value}"):
+            # Disable "Prompt Options" button if options weren't configured
+            is_options_button = placeholder_value == placeholder.PROMPT_OPTIONS
+            button_disabled = is_options_button and not options_configured
+            button_help = f"Replaces '{shortcut}' with {placeholder_value}"
+            if button_disabled:
+                button_help = "⚠️ You need to configure answer options first. Go back to the Options page to set them up."
+            
+            if cols[i].button(button_label, key=button_key, use_container_width=True, disabled=button_disabled, help=button_help):
                 st.session_state.placeholder_to_replace = {
                     "shortcut": shortcut,
                     "value": placeholder_value
@@ -177,135 +304,165 @@ if "questionnaires" in st.session_state and st.session_state.questionnaires is n
 
         st.divider()
 
-        # System prompt and main prompt section (from Basic Prompt Settings)
-        new_system_prompt = st.text_area(
-            label=system_prompt_field,
-            key=f"{system_prompt_field}{current_questionnaire_id}",
-            value=questionnaire.system_prompt,
-            help="The system prompt the model is prompted with."
-        )
-
-        change_all_system = state.create(
-            st.checkbox,
-            key=change_all_system_prompts_checkbox,
-            label="On update: change all System Prompts",
-            help="If this is ticked, all system prompts will be changed to this.",
-            initial_value=False
-        )
-
-        # Handle placeholder replacement for main prompt before widget is created
-        prompt_key = f"{prompt_field}{current_questionnaire_id}"
-        if "main_prompt_placeholder_to_replace" in st.session_state and st.session_state.main_prompt_placeholder_to_replace:
-            current_prompt_text = st.session_state.get(prompt_key, questionnaire.prompt)
-            placeholder_shortcut = st.session_state.main_prompt_placeholder_to_replace["shortcut"]
-            placeholder_value = st.session_state.main_prompt_placeholder_to_replace["value"]
-            
-            if placeholder_shortcut in current_prompt_text:
-                st.session_state[prompt_key] = current_prompt_text.replace(placeholder_shortcut, placeholder_value)
-            else:
-                st.session_state[prompt_key] = current_prompt_text + f" {placeholder_value} "
-            
-            st.session_state.main_prompt_placeholder_to_replace = None
-            st.rerun()
-
-        new_prompt = st.text_area(
-            label=prompt_field,
-            key=prompt_key,
-            value=questionnaire.prompt,
-            help="Instructions that are given to the model before the questions."
-        )
-
-        # Placeholder insertion buttons for main prompt
-        st.write("**Insert Placeholder in Main Prompt:**")
-        main_prompt_placeholders = [
-            (placeholder.PROMPT_QUESTIONS, "-P", "P", "Prompt Questions"),
-            (placeholder.PROMPT_OPTIONS, "-O", "O", "Prompt Options"),
-            (placeholder.PROMPT_AUTOMATIC_OUTPUT_INSTRUCTIONS, "-A", "A", "Automatic Output"),
-            (placeholder.JSON_TEMPLATE, "-J", "J", "JSON Template"),
-        ]
         
-        main_shortcuts_list = ", ".join([f"`{shortcut}`" for _, shortcut, _, _ in main_prompt_placeholders])
-        st.caption(f"💡 Tip: Type shortcuts {main_shortcuts_list} in the main prompt, then click the button to replace them.")
-        
-        cols_main = st.columns(len(main_prompt_placeholders))
-        for i, (placeholder_value, shortcut, char_label, description) in enumerate(main_prompt_placeholders):
-            button_key = f"btn_main_placeholder_{char_label}"
-            if cols_main[i].button(description, key=button_key, use_container_width=True, help=f"Replaces '{shortcut}' with {placeholder_value}"):
-                st.session_state.main_prompt_placeholder_to_replace = {
-                    "shortcut": shortcut,
-                    "value": placeholder_value
-                }
-                st.rerun()
-
-        change_all_questionnaire = state.create(
-            st.checkbox,
-            key=change_all_prompts_checkbox,
-            label="On update: change all questionnaire instructions",
-            help="If this is ticked, all questionnaire instructions will be changed to this.",
-            initial_value=False
-        )
 
     # Place the corresponding output in the second column
     with col2:
         st.subheader("📄 Live Preview")
-
     # --- The Dynamic Preview Logic ---
     # This block re-runs on every widget interaction.
+    # --- The Preview Logic ---
+    # Preview only updates when "Update Prompt(s)" is clicked
         with st.container(border=True):
-            # Update temporary questionnaire with question stem
-            if "survey_options" in st.session_state:
-                survey_options = st.session_state.survey_options
-            else:
-                survey_options = None
 
-            if randomize_order_bool:
-                st.session_state.temporary_questionnaire.prepare_prompt(
+            
+            #             # Update temporary questionnaire with question stem
+            # if "survey_options" in st.session_state:
+            #     survey_options = st.session_state.survey_options
+            # else:
+            #     survey_options = None
+
+            # if randomize_order_bool:
+            #     st.session_state.temporary_questionnaire.prepare_prompt(
+            #         question_stem=question_stem_input,
+            #         answer_options=survey_options,
+            #         randomized_item_order=randomize_order_bool,
+            #     )
+            # st.session_state.base_questionnaire.prepare_prompt(
+            #     question_stem=question_stem_input,
+            #     answer_options=survey_options,
+            #     randomized_item_order=False,
+            # )
+
+            # if not randomize_order_bool:
+            #     st.session_state.temporary_questionnaire = st.session_state.base_questionnaire.duplicate()
+
+            # # Update system prompt and main prompt for preview (apply to temporary_questionnaire)
+            # st.session_state.temporary_questionnaire.system_prompt = new_system_prompt
+            # st.session_state.temporary_questionnaire.prompt = new_prompt
+            # current_system_prompt, current_prompt = st.session_state.temporary_questionnaire.get_prompt_for_questionnaire_type(QuestionnairePresentation.SEQUENTIAL)
+            # current_system_prompt = current_system_prompt.replace("\n", "  \n")
+            # current_prompt = current_prompt.replace("\n", "  \n")
+            # st.write(current_system_prompt)
+            # st.write(current_prompt)
+            
+            
+            # Initialize preview state if it doesn't exist
+            preview_key = f"preview_{current_questionnaire_id}"
+            if preview_key not in st.session_state:
+                # Create initial preview
+                if "survey_options" in st.session_state:
+                    survey_options = st.session_state.survey_options
+                else:
+                    survey_options = None
+                
+                temp_questionnaire = st.session_state.temporary_questionnaire.duplicate()
+                temp_questionnaire.prepare_prompt(
                     question_stem=question_stem_input,
                     answer_options=survey_options,
-                    randomized_item_order=randomize_order_bool,
+                    randomized_item_order=False,
                 )
-            st.session_state.base_questionnaire.prepare_prompt(
-                question_stem=question_stem_input,
-                answer_options=survey_options,
-                randomized_item_order=False,
-            )
-
-            if not randomize_order_bool:
-                st.session_state.temporary_questionnaire = st.session_state.base_questionnaire.duplicate()
-
-            # Update system prompt and main prompt for preview (apply to temporary_questionnaire)
-            st.session_state.temporary_questionnaire.system_prompt = new_system_prompt
-            st.session_state.temporary_questionnaire.prompt = new_prompt
-            current_system_prompt, current_prompt = st.session_state.temporary_questionnaire.get_prompt_for_questionnaire_type(QuestionnairePresentation.SEQUENTIAL)
-            current_system_prompt = current_system_prompt.replace("\n", "  \n")
-            current_prompt = current_prompt.replace("\n", "  \n")
-            st.write(current_system_prompt)
-            st.write(current_prompt)
+                temp_questionnaire.system_prompt = questionnaire.system_prompt
+                temp_questionnaire.prompt = questionnaire.prompt
+                current_system_prompt, current_prompt = temp_questionnaire.get_prompt_for_questionnaire_type(QuestionnairePresentation.SEQUENTIAL)
+                current_system_prompt = current_system_prompt.replace("\n", "  \n")
+                current_prompt = current_prompt.replace("\n", "  \n")
+                st.session_state[preview_key] = {
+                    "system_prompt": current_system_prompt,
+                    "prompt": current_prompt
+                }
+            
+            # Display the stored preview
+            st.write(st.session_state[preview_key]["system_prompt"])
+            st.write(st.session_state[preview_key]["prompt"])
 
     st.divider()
 
     if st.button("Update Prompt(s)", type="secondary", use_container_width=True):
+        # Get current values from text areas (they're already in session state via the keys)
+        current_system_value = st.session_state.get(system_prompt_key, new_system_prompt)
+        current_prompt_value = st.session_state.get(prompt_key, new_prompt)
+        
         if change_all_system:
             for questionnaire in st.session_state.questionnaires:
-                questionnaire.system_prompt = new_system_prompt
+                questionnaire.system_prompt = current_system_value
         else:
-            st.session_state.questionnaires[current_questionnaire_id].system_prompt = new_system_prompt
+            st.session_state.questionnaires[current_questionnaire_id].system_prompt = current_system_value
 
         if change_all_questionnaire:
             for questionnaire in st.session_state.questionnaires:
-                questionnaire.prompt = new_prompt
+                questionnaire.prompt = current_prompt_value
+        else:
+            st.session_state.questionnaires[current_questionnaire_id].prompt = current_prompt_value
+        
+        # Update the preview when Update Prompt(s) is clicked
+        preview_key = f"preview_{current_questionnaire_id}"
+        if "survey_options" in st.session_state:
+            survey_options = st.session_state.survey_options
+        else:
+            survey_options = None
+        
+        temp_questionnaire = st.session_state.temporary_questionnaire.duplicate()
+        if randomize_order_bool:
+            temp_questionnaire.prepare_prompt(
+                question_stem=question_stem_input,
+                answer_options=survey_options,
+                randomized_item_order=randomize_order_bool,
+            )
         else:
-            st.session_state.questionnaires[current_questionnaire_id].prompt = new_prompt             
+            temp_questionnaire.prepare_prompt(
+                question_stem=question_stem_input,
+                answer_options=survey_options,
+                randomized_item_order=False,
+            )
+        
+        temp_questionnaire.system_prompt = current_system_value
+        temp_questionnaire.prompt = current_prompt_value
+        preview_system_prompt, preview_prompt = temp_questionnaire.get_prompt_for_questionnaire_type(QuestionnairePresentation.SEQUENTIAL)
+        preview_system_prompt = preview_system_prompt.replace("\n", "  \n")
+        preview_prompt = preview_prompt.replace("\n", "  \n")
+        st.session_state[preview_key] = {
+            "system_prompt": preview_system_prompt,
+            "prompt": preview_prompt
+        }
+             
         st.success("Prompt(s) updated!")
+        st.rerun()
 
     if st.button("Confirm and Prepare Questionnaire", type="primary", use_container_width=True):
+        # Get survey options if they exist
+        if "survey_options" in st.session_state:
+            survey_options = st.session_state.survey_options
+        else:
+            survey_options = None
+        
+        # Get current system prompt and main prompt values
+        current_system_value = st.session_state.get(system_prompt_key, new_system_prompt)
+        current_prompt_value = st.session_state.get(prompt_key, new_prompt)
+        
         for questionnaire in st.session_state.questionnaires:
+            # Update system prompt and main prompt
+            if change_all_system:
+                questionnaire.system_prompt = current_system_value
+            else:
+                # Only update if it's the current questionnaire or if we're updating all
+                if questionnaire == st.session_state.questionnaires[current_questionnaire_id]:
+                    questionnaire.system_prompt = current_system_value
+            
+            if change_all_questionnaire:
+                questionnaire.prompt = current_prompt_value
+            else:
+                # Only update if it's the current questionnaire or if we're updating all
+                if questionnaire == st.session_state.questionnaires[current_questionnaire_id]:
+                    questionnaire.prompt = current_prompt_value
+            
+            # Update question stem and answer options
             questionnaire.prepare_prompt(
                 question_stem=question_stem_input,
                 answer_options=survey_options,
                 randomized_item_order=randomize_order_bool,
             )
         st.success("Changed the prompts!")
-        st.switch_page("pages/02_Option_Prompt.py")
+        st.switch_page("pages/03_Inference_Setting.py")
 else:
     st.warning("No data found. Please upload a CSV file on the 'Start Page' first.")

src/pages/04_Final_Overview.py

@@ -8,6 +8,7 @@ import queue
 import time
 import threading
 import asyncio
+import pandas as pd
 
 import logging
 
@@ -147,20 +148,39 @@ with col_prompt_display:
     selected_method_name, selected_questionnaire_type = survey_method_options[survey_method_display]
 
     with st.container(border=True):
-        current_system_prompt, current_prompt = questionnaires.get_prompt_for_questionnaire_type(selected_questionnaire_type)
-        current_system_prompt = current_system_prompt.replace("\n", "  \n")
-        current_prompt = current_prompt.replace("\n", "  \n")
-        st.write(current_system_prompt)
-        st.write(current_prompt)
-
-model_name = state.create(
-    st.text_input,
-    "save_file",
-    "Save File",
-    # initial_value="meta-llama/Llama-3.1-70B-Instruct",
-    # placeholder="meta-llama/Llama-3.1-70B-Instruct",
-    help="The save file to write your results to. Should be a csv file.",
-)
+        # For single item mode, show multiple previews (up to 3 items)
+        if selected_questionnaire_type == QuestionnairePresentation.SINGLE_ITEM:
+            num_questions = len(questionnaires._questions)
+            num_previews = min(3, num_questions)  # Show up to 3 previews
+            
+            if num_previews > 1:
+                st.write(f"**Preview of first {num_previews} items:**")
+            else:
+                st.write("**Preview:**")
+            
+            for i in range(num_previews):
+                if num_previews > 1:
+                    st.write(f"**Item {i+1}:**")
+                
+                current_system_prompt, current_prompt = questionnaires.get_prompt_for_questionnaire_type(
+                    selected_questionnaire_type, 
+                    item_id=i
+                )
+                current_system_prompt = current_system_prompt.replace("\n", "  \n")
+                current_prompt = current_prompt.replace("\n", "  \n")
+                st.write(current_system_prompt)
+                st.write(current_prompt)
+                
+                # Add separator between items (except for the last one)
+                if i < num_previews - 1:
+                    st.divider()
+        else:
+            # For battery and sequential, show single preview as before
+            current_system_prompt, current_prompt = questionnaires.get_prompt_for_questionnaire_type(selected_questionnaire_type)
+            current_system_prompt = current_system_prompt.replace("\n", "  \n")
+            current_prompt = current_prompt.replace("\n", "  \n")
+            st.write(current_system_prompt)
+            st.write(current_prompt)
 
 
 if st.button("Confirm and Run Questionnaire", type="primary", use_container_width=True):
@@ -276,14 +296,47 @@ if st.button("Confirm and Run Questionnaire", type="primary", use_container_widt
     except queue.Empty:
         st.error("Could not retrieve result from the asynchronous task.")
 
-    st.success("Finished inferencing! Saving results...")
+    st.success("Finished inferencing!")
 
     responses = raw_responses(final_output)
 
     df = create_one_dataframe(responses)
 
-    st.dataframe(df)
+    # Store the dataframe in session state for saving later
+    st.session_state.results_dataframe = df
+    st.session_state.inference_completed = True
 
-    df.to_csv(st.session_state.save_file, index=False)
+    st.dataframe(df)
 
-    st.success(f"File saved to {st.session_state.save_file}!")
+# Show save button if inference is completed
+if "inference_completed" in st.session_state and st.session_state.inference_completed:
+    st.divider()
+    st.subheader("💾 Save Results")
+    
+    # Text input for filename
+    if "save_filename" not in st.session_state:
+        st.session_state.save_filename = "questionnaire_results.csv"
+    
+    save_filename = st.text_input(
+        "Save File",
+        value=st.session_state.save_filename,
+        key="save_filename_input",
+        help="Enter the filename for the results. Should be a CSV file (e.g., results.csv)."
+    )
+    
+    # Ensure filename ends with .csv
+    if save_filename and not save_filename.endswith('.csv'):
+        save_filename = save_filename + '.csv'
+    
+    # Convert dataframe to CSV string for download
+    csv = st.session_state.results_dataframe.to_csv(index=False)
+    
+    st.download_button(
+        label="Save Results",
+        data=csv,
+        file_name=save_filename if save_filename else "questionnaire_results.csv",
+        mime="text/csv",
+        type="primary",
+        use_container_width=True,
+        help="Click to save the results to your computer. You can choose the directory and filename in the save dialog."
+    )

src/surveyGenGui.egg-info/PKG-INFO

@@ -1,3 +0,0 @@
-Metadata-Version: 2.4
-Name: surveyGenGui
-Version: 0.0.1

src/surveyGenGui.egg-info/SOURCES.txt

@@ -1,17 +0,0 @@
-pyproject.toml
-src/Start_Page.py
-src/__init__.py
-src/gui_elements/__init__.py
-src/gui_elements/output_manager.py
-src/gui_elements/paginator.py
-src/gui_elements/stateful_widget.py
-src/pages/01_Basic_Prompt_Settings.py
-src/pages/02_Option_Prompt.py
-src/pages/03_Prepare_Prompts.py
-src/pages/04_Inference_Setting.py
-src/pages/05_Final_Overview.py
-src/pages/__init__.py
-src/surveyGenGui.egg-info/PKG-INFO
-src/surveyGenGui.egg-info/SOURCES.txt
-src/surveyGenGui.egg-info/dependency_links.txt
-src/surveyGenGui.egg-info/top_level.txt

src/surveyGenGui.egg-info/dependency_links.txt

@@ -1 +0,0 @@
-

src/surveyGenGui.egg-info/top_level.txt

@@ -1,4 +0,0 @@
-Start_Page
-__init__
-gui_elements
-pages