Update app.py

app.py

@@ -29,20 +29,25 @@ def evaluate(
     
     df_result = pd.DataFrame([[instruction, input, None, None],],columns=["instruction" ,"input", "output", "total_tokens"])
 
-    response = openai.ChatCompletion.create(
-      model="gpt-3.5-turbo",
-      # model="gpt-4",
-      messages=[
-        {"role": "system", "content": "You are a useful assistant."},
-        {"role": "user", "content": message },
-      ],
-      n=1,
-      presence_penalty=-2.0,
-      max_tokens=512
-    )
+    if os.environ["PROMPT2"] in message:
+        response = openai.ChatCompletion.create(
+          model="gpt-3.5-turbo",
+          # model="gpt-4",
+          messages=[
+            {"role": "system", "content": "You are a useful assistant."},
+            {"role": "user", "content": message },
+          ],
+          n=1,
+          presence_penalty=-2.0,
+          max_tokens=512
+        )
+    
+        df_result.iloc[0, 2] = response["choices"][0]["message"]["content"]
+        df_result.iloc[0, 3] = response["usage"]["total_tokens"]
 
-    df_result.iloc[0, 2] = response["choices"][0]["message"]["content"]
-    df_result.iloc[0, 3] = response["usage"]["total_tokens"]
+    else:
+        df_result.iloc[0, 2] = ""
+        df_result.iloc[0, 3] = 0      
     
     now = datetime.datetime.now()
     file_name = f'chatgpt3/test_{now.strftime("%Y%m%d_%H%M%S")}.csv'