additional prompt

openai_api.py

@@ -16,8 +16,12 @@ def sentiment(text):
     # Create a prompt for the model
     prompt = f"""You are trained to analyze and detect the sentiment of the given text.
     If you are unsure of an answer, you can say "not sure" and recommend the user review manually.
-    Analyze the following text and determine if the sentiment is: POSITIVE, NEGATIVE, NEUTRAL or MIXED.
-    {text}"""
+    Analyze the following text and determine if the sentiment is: POSITIVE, NEGATIVE or NEUTRAL.
+    Reply in single word.
+    Examples
+    Input: dress was beautiful. Output: POSITIVE
+    Input: pizza had weird smell. Output: NEGATIVE
+    Input: {text}. Output:"""
 
     # Call the OpenAI API to generate a response
     response = client.chat.completions.create(
@@ -26,11 +30,11 @@ def sentiment(text):
             {"role": "system", "content": "You are a helpful assistant."},
             {"role": "user", "content": prompt}
         ],
-        max_tokens=1,  # Limit response to a single word
+        # max_tokens=1,  # Limit response to a single word
         temperature=0  # Keep response consistent
     )
     print(response)
     # Extract the sentiment from the response
-    sentiment = response.choices[0].message.content.strip().lower()
+    sentiment = response.choices[0].message.content.strip()
 
     return sentiment