Update chatgpt_api.py

chatgpt_api.py

@@ -17,4 +17,19 @@ def get_chatgpt_response(input_text):
     response_json = response.json()
     output_text = response_json["choices"][0]["message"]["content"]
 
+    return output_text
+    
+def get_chatgpt3_response(input_text):
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {OPENAI_API_KEY}"
+    }
+    data = {
+        "model": "gpt-3",
+        "messages": [{"role": "user", "content": input_text}]
+    }
+    response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=data)
+    response_json = response.json()
+    output_text = response_json["choices"][0]["message"]["content"]
+
     return output_text