fixed output generation prompt

app.py

@@ -786,11 +786,16 @@ def generate_synthetic_outputs(
     # Format output variable information for the prompt
     output_vars_text = "\n".join(
         [
-            f"- {var['name']}: {var['description']} (Type: {var['type']}) {'Options: '+str(var['options']) if var.get('options') else ''}"
+            f"Key: {var['name']}: {var['description']} (Type: {var['type']}) {'Options: '+str(var['options']) if var.get('options') else ''}"
             for var in output_vars
         ]
     )
 
+    output_format = "{"
+    for var in output_vars:
+        output_format += f'"{var["name"]}": output, '
+    output_format = output_format.rstrip(", ") + "}"
+
     results = []
 
     # Create a progress bar
@@ -820,18 +825,15 @@ PROMPT USED:
 REQUIRED OUTPUT VARIABLES:
 {output_vars_text}
 
-Generate realistic output data for these variables. Return ONLY a JSON object with the output variables:
-{{
-  "output_variable_1": value1,
-  "output_variable_2": value2,
-  ...
-}}
+Generate realistic output data for these variables. Return ONLY a JSON object with the below format, using the names of the required output variables as keys:
+{output_format}
 
 Use appropriate data types for each variable. Return ONLY the JSON object with no additional text or explanation.
 The response must be valid JSON that can be parsed directly.
 """
 
             output_data = None
+
             for attempt in range(max_retries):
                 try:
                     response = client.chat.completions.create(
@@ -937,7 +939,6 @@ The response must be valid JSON that can be parsed directly.
 
     return results
 
-
 def suggest_variable_values_from_kb(
     variable_name, variable_type, knowledge_base, client, model="gpt-3.5-turbo"
 ):