Update app.py

app.py

@@ -40,7 +40,12 @@ def respond(
     ):
         token = message.choices[0].delta.content
         response += token
-        yield response
+
+    # Validate the response format
+    if len(response) == 13 and all(c.isdigit() or c in 'CW' for c in response):
+        return response
+    else:
+        return "Invalid response format"
 
 
 """
@@ -50,43 +55,38 @@ demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(
-            value="You are tasked with labeling text data based on both emotion temperature and text type categories. The final output must be a 13-character code that consists of the following structure",
-            label="""
-1. Emotion Temperature Code (2 characters):
-   - If the emotion is purely Cold: Use CC
-   - If the emotion is purely Warm: Use WW
-   - If the emotion is purely Hot: Use HH
-   - If the emotion is a mix, use one of the following:
-     - Cold and Warm: Use CW
-     - Warm and Hot: Use WH
-     - Cold and Hot: Use CH
-
-2. Text Type Codes (next 9 digits):
-   Assign a digit for each of the following categories based on the presence in the text. Use 0 for categories not applicable:
-   1: Toxic
-   2: Appreciation
-   3: Constructive Criticism
-   4: Genuine Questions
-   5: Advice/Suggestions
-   6: Requests
-   7: Spam
-   8: Off-Topic
-   9: Engagement Boosters
-
-3. Special Categories (last 2 digits):
-   If the text is Neutral/General: Set the 10th digit to 1; otherwise, set it to 0.
-   If the text contains Hate: Set the last digit (11th) to 1; otherwise, set it to 0.
-
-Example:
-For the text "I love your videos but still something is missing":
-   - Emotion: Cold and Warm (CW)
-   - Types Detected: 2 (Appreciation), 3 (Constructive Criticism), 5 (Advice/Suggestions)
-   - Special Categories: Neutral/General (set the 10th digit to 1), no Hate
-
-The output would be: CW02305000010
-
-Output Format:
-Always return a 13-character code following this structure.""",
+            value="You are tasked with labeling text data based on both emotion temperature and text type categories. The final output must be a 13-character code that consists of the following structure:\n\n"
+                  "1. Emotion Temperature Code (2 characters):\n"
+                  "   - If the emotion is purely Cold: Use CC\n"
+                  "   - If the emotion is purely Warm: Use WW\n"
+                  "   - If the emotion is purely Hot: Use HH\n"
+                  "   - If the emotion is a mix, use one of the following:\n"
+                  "     - Cold and Warm: Use CW\n"
+                  "     - Warm and Hot: Use WH\n"
+                  "     - Cold and Hot: Use CH\n\n"
+                  "2. Text Type Codes (next 9 digits):\n"
+                  "   Assign a digit for each of the following categories based on the presence in the text. Use 0 for categories not applicable:\n"
+                  "   1: Toxic\n"
+                  "   2: Appreciation\n"
+                  "   3: Constructive Criticism\n"
+                  "   4: Genuine Questions\n"
+                  "   5: Advice/Suggestions\n"
+                  "   6: Requests\n"
+                  "   7: Spam\n"
+                  "   8: Off-Topic\n"
+                  "   9: Engagement Boosters\n\n"
+                  "3. Special Categories (last 2 digits):\n"
+                  "   If the text is Neutral/General: Set the 10th digit to 1; otherwise, set it to 0.\n"
+                  "   If the text contains Hate: Set the last digit (11th) to 1; otherwise, set it to 0.\n\n"
+                  "Example:\n"
+                  "For the text 'I love your videos but still something is missing':\n"
+                  "   - Emotion: Cold and Warm (CW)\n"
+                  "   - Types Detected: 2 (Appreciation), 3 (Constructive Criticism), 5 (Advice/Suggestions)\n"
+                  "   - Special Categories: Neutral/General (set the 10th digit to 1), no Hate\n\n"
+                  "The output would be: CW02305000010\n\n"
+                  "Output Format:\n"
+                  "Always return a 13-character code following this structure.",
+            label="Instructions",
             lines=10,
         ),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),