Merge pull request #3 from Haste171/async

Add Async

Author: Haste171

.gitignore

@@ -0,0 +1 @@
+.idea/

gptzero/api.py

@@ -1,7 +1,9 @@
 import os
+import aiohttp
 import requests
 
 
+
 class GPTZeroAPI:
     def __init__(self, api_key):
         self.api_key = api_key
@@ -31,3 +33,35 @@ def file_predict(self, file_path):
         }
         response = requests.post(url, headers=headers, files=files)
         return response.json()
+
+
+class AsyncGPTZeroAPI:
+    def __init__(self, api_key):
+        self.api_key = api_key
+        self.base_url = 'https://api.gptzero.me/v2/predict'
+
+
+    async def text_predict(self, session: aiohttp.ClientSession, document):
+        url = f'{self.base_url}/text'
+        headers = {
+            'accept': 'application/json',
+            'X-Api-Key': self.api_key,
+            'Content-Type': 'application/json'
+        }
+        data = {
+            'document': document
+        }
+        async with session.get(url, headers=headers, json=data, ssl=False) as response:
+            return await response.json()
+
+    async def file_predict(self, session: aiohttp.ClientSession, file_path):
+        url = f'{self.base_url}/files'
+        headers = {
+            'accept': 'application/json',
+            'X-Api-Key': self.api_key
+        }
+        files = {
+            'files': (os.path.basename(file_path), open(file_path, 'rb'))
+        }
+        async with session.get(url, headers=headers, files=files, ssl=False) as response:
+            return await response.json()

requirements.txt

@@ -0,0 +1,14 @@
+aiohttp==3.8.4
+aiosignal==1.3.1
+async-timeout==4.0.2
+attrs==22.2.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+frozenlist==1.3.3
+idna==3.4
+multidict==6.0.4
+urllib3==1.26.15
+yarl==1.8.2
+
+requests~=2.28.2
+setuptools~=65.5.1

setup.py

@@ -2,13 +2,14 @@
 
 setup(
     name='gptzero',
-    version='0.1.1',
-    description='Python wrapper for the GPTZero API',
-    author='David Peterson',
-    author_email='dapanon@proton.me',
-    url='https://github.com/haste171/gptzero',
+    version='0.1.2',
+    description='Asynchronous Python wrapper for the GPTZero API',
+
+    author="David Peterson",
+    author_email="dapanon@proton.me",
+    url='https://github.com/Haste171/gptzero/',
     packages=find_packages(),
-    install_requires=['requests'],
+    install_requires=['aiohttp'],
     classifiers=[
         'Development Status :: 3 - Alpha',
         'Intended Audience :: Developers',
@@ -19,6 +20,7 @@
         'Programming Language :: Python :: 3.8',
         'Programming Language :: Python :: 3.9',
         'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
     ],
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',