Update app.py

app.py

@@ -10,8 +10,8 @@ import numpy as np
 import streamlit as st
 from openai import OpenAI
 import os
-import sys
 from dotenv import load_dotenv
+import random
 
 load_dotenv()
 
@@ -101,21 +101,24 @@ if task_choice == "Data Generation":
         ]
     else:
         few_shot_examples = []
-###here!!!
+
     # Ask the user how many examples they need
     num_to_generate = st.number_input("How many examples to generate?", min_value=1, max_value=50, value=10)
 
+    # User prompt text field
+    user_prompt = st.text_area("Enter your prompt to guide example generation", "")
+
     # System prompt generation
     system_prompt = f"You are a professional {classification_type.lower()} expert. Your role is to generate data for {domain}.\n\n"
     if few_shot_examples:
         system_prompt += "Use the following few-shot examples as a reference:\n"
         for example in few_shot_examples:
             system_prompt += f"Example: {example['content']} \n Label: {example['label']}\n"
-    system_prompt += f"Generate {num_to_generate} examples.\n"
+    system_prompt += f"Generate {num_to_generate} unique examples with diverse phrasing.\n"
     system_prompt += f"Each example should have between {min_words} and {max_words} words.\n"
-    system_prompt +=  f"Think step by step while generating the examples and use the labels specified."
-    ####
-    #system_prompt += f"Generate {num_to_generate} examples."
+    system_prompt += f"Use the labels specified: {', '.join(labels)}.\n"
+    if user_prompt:
+        system_prompt += f"Additional instructions: {user_prompt}\n"
 
     st.write("System Prompt:")
     st.code(system_prompt)
@@ -149,3 +152,4 @@ else:
     # Data labeling workflow (for future implementation based on classification)
     st.write("Data Labeling functionality will go here.")
 
+