Update app.py

app.py

@@ -78,14 +78,18 @@ def generate_response(i, messages):
         completion = openai.ChatCompletion.create(
             model=model,
             messages=messages,
-            temperature=temperature,
             max_tokens=max_tokens,
-            top_p=top_p
+            temperature=temperature,
+            top_p=top_p,
+            frequency_penalty=0, # You can adjust this value
+            presence_penalty=0, # You can adjust this value
+            stop=None # You can specify stop sequences if needed
         )
 
-        response = completion.choices[0].message.content
-        prompt_tokens = completion.usage['prompt_tokens']
-        response_tokens = completion.usage['total_tokens'] - prompt_tokens
+        # Extracting the response from the completion
+        response = completion['choices'][0]['message']['content']
+        prompt_tokens = completion['usage']['prompt_tokens']
+        response_tokens = completion['usage']['total_tokens'] - prompt_tokens
         word_count = len(re.findall(r'\w+', response))
 
         return (i, response, prompt_tokens, response_tokens, word_count, None)