Add files via upload

gpt_engine.py

@@ -0,0 +1,4 @@
+from .pygpt4 import ask_gpt
+
+def ask_gpt(api_key, prompt):
+    return ask_gpt(api_key, prompt)

pygpt4.py

@@ -0,0 +1,22 @@
+import requests
+
+def ask_gpt(api_key, prompt, max_tokens=50):
+    api_url = 'https://api.openai.com/v1/engines/davinci-codex/completions'
+
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {api_key}'
+    }
+
+    data = {
+        'prompt': prompt,
+        'max_tokens': max_tokens,
+        'temperature': 0.7,
+    }
+
+    response = requests.post(api_url, json=data, headers=headers)
+
+    if response.status_code == 200:
+        return response.json()['choices'][0]['text']
+    else:
+        raise Exception(f"ERROR: {response.status_code}, {response.json()['error']['message']}")

setup.py

@@ -0,0 +1,10 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='pygpt4',
+    version='0.1',
+    packages=find_packages(),
+    install_requires=[
+        'requests',
+    ],
+)