Update app.py

app.py

@@ -21,7 +21,7 @@ add_row = st.sidebar.button("Add row")
 reset = st.sidebar.button("Reset")
 followup_message = st.sidebar.text_area("Edit Message:")
 generate_all_edits = st.sidebar.button("Generate All Edits")
-model = st.sidebar.selectbox("Model:", ["gpt-4", "gpt-4-1106-preview"])
+model = st.sidebar.selectbox("Model:", ["gpt-4-1106-preview", "gpt-4"])
 temperature = st.sidebar.slider("Temperature:", 0.0, 1.0, 0.6, step=0.01)
 max_tokens = st.sidebar.number_input("Max Tokens:", min_value=1, max_value=8192, value=1000, step=1)
 top_p = st.sidebar.slider("Top P:", 0.0, 1.0, 1.0, step=0.01)
@@ -78,18 +78,14 @@ def generate_response(i, messages):
         completion = openai.ChatCompletion.create(
             model=model,
             messages=messages,
-            max_tokens=max_tokens,
             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
+            max_tokens=max_tokens,
+            top_p=top_p
         )
 
-        # 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
+        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)