chore: prompt

app.py

@@ -29,33 +29,28 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
 
 system_prompt = """
-You are a customer support officer. Your task is to extract a transaction ID from the USER INPUT and determine its transaction type.
+You are a customer support officer. Extract the transaction ID from the USER INPUT and determine its type.
 
-To extract the transaction ID, follow these steps:
+Extraction rules:
+- Look for words starting with 'payout' or 'payment'.
+- The next character must be a dash ('-').
+- There must be at least one digit or character after the dash.
+- The transaction ID must appear exactly in the USER INPUT.
+- If found, set found to true; otherwise, set found to false.
 
-1.For each word in the USER INPUT, check if it starts with 'payout' or 'payment'. If yes, continue to Step 2; if not, go to Step 7.
-2.Check if the character immediately after the prefix ('payout' or 'payment') is a dash ('-'). If yes, continue to Step 3; if not, go to Step 7.
-3.Check if there is at least one digit or character after the dash. If yes, continue to Step 4; if not, go to Step 7.
-4.Extract the transaction ID from the USER INPUT and continue to Step 5.
-5.Verify that the extracted transaction ID appears exactly in the USER INPUT. If yes, continue to Step 6; if not, go to Step 7.
-6.Mark the transaction ID as valid and set the flag found to true.
-7.If no valid transaction ID is found, set the flag found to false.
-
-To determine the transaction type:
-
-1.If the transaction ID starts with 'payout', the type is payout.
-2.If the transaction ID starts with 'payment', the type is payment.
+Type rules:
+- If the transaction ID starts with 'payout', type is payout.
+- If it starts with 'payment', type is payment.
 
 ===>USER INPUT BEGINS
 {input}
 <===USER INPUT ENDS
 
-Respond in valid JSON with the following fields:
-
+Respond in valid JSON with these fields:
 found: (boolean) Whether a valid transaction ID was found.
 transaction_id: (string, if found) The extracted transaction ID.
 transaction_type: (string, if found) The transaction type in lowercase.
-justification: (string) An explanation of how the transaction ID and type were determined. If no valid transaction ID is found, do not fabricate an explanation.
+justification: (string) Explain how you determined the transaction ID and type. If not found, do not fabricate an explanation.
 Return only valid JSON and nothing else.
 """