Update app10.py

app10.py

@@ -128,6 +128,7 @@ if task_choice == "Data Generation":
                 st.error("An error occurred during generation.")
                 st.error(f"Details: {e}")
 
+    
 elif task_choice == "Data Labeling":
     # Labeling logic
     labeling_type = st.selectbox(
@@ -158,14 +159,31 @@ elif task_choice == "Data Labeling":
 
     if st.button("Classify Text"):
         if text_to_classify:
-            labeling_prompt = (
-                f"You are an expert in {labeling_type.lower()} classification. Classify this text using: {', '.join(labels)}.\n\n"
+            # Construct the labeling prompt
+            labeling_prompt_template = PromptTemplate(
+                input_variables=["labeling_type", "labels", "few_shot_examples", "text_to_classify"],
+                template=(
+                    "You are an expert in {labeling_type} classification. "
+                    "Classify the following text using: {labels}.\n\n"
+                    "{few_shot_examples}\n"
+                    "Classify this: {text_to_classify}"
+                )
             )
+
+            # Prepare few-shot examples for the prompt
+            few_shot_examples_text = ""
             if few_shot_labeling_examples:
-                labeling_prompt += "Example classifications:\n"
+                few_shot_examples_text += "Example classifications:\n"
                 for ex in few_shot_labeling_examples:
-                    labeling_prompt += f"Text: {ex['content']} - Label: {ex['label']}\n"
-            labeling_prompt += f"\nClassify this: {text_to_classify}"
+                    few_shot_examples_text += f"Text: {ex['content']} - Label: {ex['label']}\n"
+
+            # Format the prompt with the user's input
+            labeling_prompt = labeling_prompt_template.format(
+                labeling_type=labeling_type.lower(),
+                labels=", ".join(labels),
+                few_shot_examples=few_shot_examples_text.strip(),
+                text_to_classify=text_to_classify
+            )
 
             with st.spinner("Classifying..."):
                 st.session_state.messages.append({"role": "system", "content": labeling_prompt})
@@ -178,7 +196,9 @@ elif task_choice == "Data Labeling":
                         max_tokens=3000,
                     )
                     labeling_response = st.write_stream(stream)
-                    st.write("Label:", labeling_response)
+                    # Format response to match desired output
+                    formatted_response = f"Label: {labeling_response}"
+                    st.write(formatted_response)
                 except Exception as e:
                     st.error("An error occurred during classification.")
                     st.error(f"Details: {e}")