Update app.py

app.py

@@ -361,7 +361,8 @@ def generate_synthetic_inputs(template_spec, num_samples=1, max_retries=3):
             f"- {var['name']}: {var['description']} (Type: {var['type']})"
             + (
                 f", Min: {var.get('min', 'N/A')}, Max: {var.get('max', 'N/A')}"
-                if var["type"] in ["string", "int", "float"]
+                if var["type"]
+                in ["string", "int", "float", "categorical"]  # Added categorical here
                 else ""
             )
             + (f", Options: {var['options']}" if var.get("options") else "")
@@ -370,26 +371,27 @@ def generate_synthetic_inputs(template_spec, num_samples=1, max_retries=3):
     )
 
     prompt = f"""
-You are a synthetic data generator. Generate {num_samples} realistic sample(s) for the following input variables:
+    You are a synthetic data generator. Generate {num_samples} realistic sample(s) for the following input variables:
 
-{input_vars_text}
-
-Return the data as a JSON array of objects, where each object contains values for all input variables.
-Each object should follow this structure:
-{{
-  "variable_name_1": value1,
-  "variable_name_2": value2,
-  ...
-}}
+    {input_vars_text}
 
-Make sure to:
-1. Use appropriate data types (strings, numbers, booleans)
-2. Stay within min/max constraints
-3. Only use provided options for categorical variables
-4. Generate realistic and diverse values
-5. Return ONLY the JSON array with no additional text or explanation
-6. The response must be valid JSON that can be parsed directly
-"""
+    Return the data as a JSON array of objects, where each object contains values for all input variables.
+    Each object should follow this structure:
+    {{
+    "variable_name_1": value1,
+    "variable_name_2": value2,
+    ...
+    }}
+
+    Make sure to:
+    1. Use appropriate data types (strings, numbers, booleans)
+    2. Stay within min/max constraints
+    3. Only use provided options for categorical variables
+    4. For categorical variables with min > 1 or max > 1, return an array of selected options with length between min and max
+    5. Generate realistic and diverse values
+    6. Return ONLY the JSON array with no additional text or explanation
+    7. The response must be valid JSON that can be parsed directly
+    """
 
     for attempt in range(max_retries):
         try:
@@ -671,7 +673,11 @@ with tab1:
     # Add option to either upload a template or create a new one
     setup_option = st.radio(
         "Choose how to start your project",
-        options=["Upload existing template", "Create new template from documents"],
+        options=[
+            "Upload existing template",
+            "Create new template from documents",
+            "Create an empty template",
+        ],
         index=1,
     )
 
@@ -702,12 +708,8 @@ with tab1:
                         "Template loaded! Go to the 'Edit Template' tab to customize it."
                     )
 
-    if (
-        setup_option == "Create new template from documents"
-        or setup_option == "Upload existing template"
-        and not uploaded_template
-    ):
-        # Step 1: Upload Knowledge Base (existing code)
+    elif setup_option == "Create new template from documents":
+        # Step 1: Upload Knowledge Base
         st.subheader("Step 1: Upload Knowledge Base")
         uploaded_files = st.file_uploader(
             "Upload documents to use as knowledge base",
@@ -762,6 +764,54 @@ with tab1:
             else:
                 st.warning("Please provide instructions first")
 
+    elif setup_option == "Create an empty template":
+        st.subheader("Create Empty Template")
+        st.info(
+            "This option creates a minimal template that you can customize in the 'Edit Template' tab."
+        )
+
+        # Optional: Allow setting a name and description for the template
+        template_name = st.text_input("Template Name", value="Custom Template")
+        template_description = st.text_area(
+            "Template Description", value="A custom template created from scratch"
+        )
+
+        if st.button("Create Empty Template"):
+            # Create a minimal template structure
+            st.session_state.template_spec = {
+                "name": template_name,
+                "version": "1.0.0",
+                "description": template_description,
+                "input": [
+                    {
+                        "name": "input_1",
+                        "description": "First input variable",
+                        "type": "string",
+                        "min": 1,
+                        "max": 100,
+                    }
+                ],
+                "output": [
+                    {
+                        "name": "output_1",
+                        "description": "Generated output",
+                        "type": "string",
+                        "min": 10,
+                        "max": 1000,
+                    }
+                ],
+                "prompt": "Based on the following information:\n{input_1}\n\nGenerate the following output.",
+            }
+
+            st.session_state.show_template_editor = True
+            st.success(
+                "Empty template created! Go to the 'Edit Template' tab to customize it."
+            )
+
+            # Optional: Initialize an empty knowledge base
+            if "knowledge_base" not in st.session_state:
+                st.session_state.knowledge_base = ""
+
 with tab2:
     if st.session_state.show_template_editor and st.session_state.template_spec:
         st.header("Template Editor")
@@ -891,6 +941,23 @@ with tab2:
                                 if opt.strip()
                             ]
 
+                            # Add min and max for categorical variables
+                            col_min, col_max = st.columns(2)
+                            with col_min:
+                                input_var["min"] = st.number_input(
+                                    "Min selections",
+                                    value=int(input_var.get("min", 0)),
+                                    min_value=0,
+                                    key=f"input_cat_min_{i}",
+                                )
+                            with col_max:
+                                input_var["max"] = st.number_input(
+                                    "Max selections",
+                                    value=int(input_var.get("max", 1)),
+                                    min_value=1,
+                                    key=f"input_cat_max_{i}",
+                                )
+
                     with col3:
                         if st.button("Remove", key=f"remove_input_{i}"):
                             st.session_state.template_spec["input"].pop(i)
@@ -975,6 +1042,23 @@ with tab2:
                                 if opt.strip()
                             ]
 
+                            # Add min and max for categorical variables
+                            col_min, col_max = st.columns(2)
+                            with col_min:
+                                output_var["min"] = st.number_input(
+                                    "Min selections",
+                                    value=int(output_var.get("min", 0)),
+                                    min_value=0,
+                                    key=f"output_cat_min_{i}",
+                                )
+                            with col_max:
+                                output_var["max"] = st.number_input(
+                                    "Max selections",
+                                    value=int(output_var.get("max", 1)),
+                                    min_value=1,
+                                    key=f"output_cat_max_{i}",
+                                )
+
                     with col3:
                         if st.button("Remove", key=f"remove_output_{i}"):
                             st.session_state.template_spec["output"].pop(i)
@@ -1048,12 +1132,29 @@ with tab3:
 
             elif var_type == "categorical":
                 options = input_var.get("options", [])
+                min_selections = input_var.get("min", 1)
+                max_selections = input_var.get("max", 1)
+
                 if options:
-                    st.session_state.user_inputs[var_name] = st.selectbox(
-                        f"Select value for {var_name}",
-                        options=options,
-                        key=f"use_{var_name}",
-                    )
+                    if min_selections == 1 and max_selections == 1:
+                        # Single selection
+                        st.session_state.user_inputs[var_name] = st.selectbox(
+                            f"Select value for {var_name}",
+                            options=options,
+                            key=f"use_{var_name}",
+                        )
+                    else:
+                        # Multi-selection
+                        st.session_state.user_inputs[var_name] = st.multiselect(
+                            f"Select {min_selections}-{max_selections} values for {var_name}",
+                            options=options,
+                            default=(
+                                options[:min_selections]
+                                if len(options) >= min_selections
+                                else options
+                            ),
+                            key=f"use_{var_name}",
+                        )
                 else:
                     st.warning(f"No options defined for {var_name}")