Rewrite request

app.py

@@ -1,22 +1,44 @@
 import gradio as gr
 import openai
+import os
+import requests
 
-openai.api_key = "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"
+openai.api_key = "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"
+# openai.api_key = os.getenv("OPENAI_API_KEY")
 
 default_system_message = {"role": "system", "content": "You are a brilliant, helpful assistant, always providing answers to the best of your knowledge. If you are unsure of the answer, you indicate it to the user."}
 
 def get_completion(model, user_message, message_history, chatlog_history, temperature, maximum_length, top_p, frequency_penalty, presence_penalty):
     new_history_row = {"role": "user", "content": user_message}
     updated_message_history = message_history + [new_history_row]
-    completion = openai.ChatCompletion.create(
-        model=model,
-        messages=updated_message_history,
-        temperature=temperature,
-        max_tokens=maximum_length,
-        top_p=top_p,
-        frequency_penalty=frequency_penalty,
-        presence_penalty=presence_penalty,
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai.api_key}",
+    }
+    payload = {
+        "model":model,
+        "messages":updated_message_history,
+        "temperature":temperature,
+        "max_tokens":maximum_length,
+        "top_p":top_p,
+        "frequency_penalty":frequency_penalty,
+        "presence_penalty":presence_penalty,
+    }
+    completion = requests.post(
+        "https://api.openai.com/v1/chat/completions",
+        headers=headers,
+        json=payload,
     )
+    completion = completion.json()
+    # completion = openai.ChatCompletion.create(
+    #     model=model,
+    #     messages=updated_message_history,
+    #     temperature=temperature,
+    #     max_tokens=maximum_length,
+    #     top_p=top_p,
+    #     frequency_penalty=frequency_penalty,
+    #     presence_penalty=presence_penalty,
+    # )
     assistant_message = completion["choices"][0]["message"]["content"]
     new_history_row = {"role": "assistant", "content": assistant_message}
     updated_message_history = updated_message_history + [new_history_row]
@@ -31,15 +53,34 @@ def retry_completion(model, message_history, chatlog_history, temperature, maxim
     updated_chatlog_history = chatlog_history[:-1]
     # delete latest assistant message from message_history
     updated_message_history = message_history[:-1]
-    completion = openai.ChatCompletion.create(
-        model=model,
-        messages=updated_message_history,
-        temperature=temperature,
-        max_tokens=maximum_length,
-        top_p=top_p,
-        frequency_penalty=frequency_penalty,
-        presence_penalty=presence_penalty,
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai.api_key}",
+    }
+    payload = {
+        "model":model,
+        "messages":updated_message_history,
+        "temperature":temperature,
+        "max_tokens":maximum_length,
+        "top_p":top_p,
+        "frequency_penalty":frequency_penalty,
+        "presence_penalty":presence_penalty,
+    }
+    completion = requests.post(
+        "https://api.openai.com/v1/chat/completions",
+        headers=headers,
+        json=payload,
     )
+    completion = completion.json()
+    # completion = openai.ChatCompletion.create(
+    #     model=model,
+    #     messages=updated_message_history,
+    #     temperature=temperature,
+    #     max_tokens=maximum_length,
+    #     top_p=top_p,
+    #     frequency_penalty=frequency_penalty,
+    #     presence_penalty=presence_penalty,
+    # )
     assistant_message = completion["choices"][0]["message"]["content"]
     new_history_row = {"role": "assistant", "content": assistant_message}
     updated_message_history = updated_message_history + [new_history_row]
@@ -76,5 +117,5 @@ with gr.Blocks(theme=theme) as app:
     retry_button.click(retry_completion, inputs=[model, message_history, chatlog_history, temperature, maximum_length, top_p, frequency_penalty, presence_penalty], outputs=[user_message, message_history, chatlog_history, chatbot, token_count])
     reset_button.click(reset_chat, inputs=[], outputs=[user_message, message_history, chatlog_history, chatbot, token_count])
 
-# app.launch()
-app.launch(auth=("admin", "C%nc6mrn8*BCwQF9HhH4CX35d7Q**eQY"))
+app.launch()
+# app.launch(auth=("admin", "C%nc6mrn8*BCwQF9HhH4CX35d7Q**eQY"))