worqhat 0.0.1__py3-none-any.whl

worqhat/__init__.py

@@ -0,0 +1,10 @@
+from .ai_models.ai_search import *
+from .ai_models.content_mod import *
+from .ai_models.image_analysis import *
+from .ai_models.image_gen import *
+from .ai_models.model_train import *
+from .ai_models.text_extract import *
+from .ai_models.text_gen import *
+from .database_management.collection import *
+from .database_management.Edit import *
+from .database_management.Read import *

worqhat/ai_models/__init__.py

@@ -0,0 +1,9 @@
+__all__ = ['ai_search', 'content_mod', 'image_analysis', 'image_gen', 'model_train', 'text_extract', 'text_gen']
+
+from .ai_search import *
+from .content_mod import *
+from .image_analysis import *
+from .image_gen import *
+from .model_train import *
+from .text_extract import *
+from .text_gen import *

worqhat/ai_models/ai_search.py

@@ -0,0 +1,41 @@
+import os
+import requests
+
+def search_ai_v2(question="", training_data="", api_key=None):
+    if question == "":
+        return "Question is incomplete. Please give a question and try again."
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/search/v2"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "question": question,
+        "training_data": training_data
+    }
+    response = requests.request("POST", url, json=payload, headers=headers)
+    return response.text
+
+def search_ai_v3(question="", training_data="", search_count=10, api_key=None):
+    if question == "":
+        return "Question is incomplete. Please give a question and try again"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/search/v3"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "question": question,
+        "training_data": training_data,
+        "search_count": search_count
+    }
+    response = requests.request("POST", url, json=payload, headers=headers)
+    return response.text

worqhat/ai_models/content_mod.py

@@ -0,0 +1,44 @@
+import requests 
+import os
+
+
+def content_moderation(text_content="",api_key=None):
+    if text_content == "":
+        return "Text content is incomplete. Please give some text and try again. "
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/moderation"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "text_content": text_content
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.text
+
+def image_moderation(image=None, api_key=None):
+    if not image or len(image) == 0:
+        return "No images found. Please try again "
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/v2/image-moderation"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+       
+    if image.startswith('http://') or image.startswith('https://'):        
+        response = requests.get(image)
+        files = [('image', ('file', response.content, 'application/octet-stream'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('image', ('file', f, 'application/octet-stream'))]
+    response = requests.post(url, files=files, headers=headers)
+    return response.text
+
+    

worqhat/ai_models/image_analysis.py

@@ -0,0 +1,94 @@
+import requests
+import os
+import json
+
+def analyze_images(images, question="", training_data="", output_type="text", stream_data=False, api_key=None):
+    if not images or len(images) == 0:
+        return "No images found. Please try again "
+
+    api_key = api_key or os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/v2/image-analysis"
+    headers = {
+        'Accept': 'application/json',
+        'Authorization': f'Bearer {api_key}'
+    }
+    payload = {
+        'output_type': output_type,
+        'stream_data': str(stream_data).lower(),
+        'question': question,
+        'training' : training_data
+    }
+    files = []
+    for image in images:
+        if image.startswith('http://') or image.startswith('https://'):
+            response = requests.get(image)
+            files.append(('images', (os.path.basename(image), response.content, 'image/jpeg')))
+        else:
+            with open(image, 'rb') as f:
+                file_content = f.read()
+                files.append(('images', (os.path.basename(image), file_content, 'image/jpeg')))
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    if stream_data:
+        for line in response.iter_lines():
+            if line:
+                if line.startswith(b'data:'):
+                    json_content = line[len(b'data:'):].decode('utf-8').strip()
+                    print(json_content)
+    else:
+        json_response = response.text
+        try:
+            json_data = json.loads(json_response)
+            print(json.dumps(json_data, indent=4))
+        except json.JSONDecodeError:
+            print(json_response)
+
+def detect_faces(image, api_key=None):
+    if not image or len(image) == 0:
+        return "No images found. Please try again "
+    api_key = api_key or os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/v2/face-detection"
+    headers = {
+        'Accept': 'application/json',
+        'Authorization': f'Bearer {api_key}'
+    }
+
+    if image.startswith('http://') or image.startswith('https://'):
+        response = requests.get(image)
+        file=[('image', (os.path.basename(image), response.content, 'image/jpeg'))]
+    else:
+        with open(image, 'rb') as f:
+            file=[('image', (os.path.basename(image), f, 'image/jpeg'))]
+    response = requests.post(url, headers=headers, files=file)
+    return response.text
+
+def compare_faces(source_image, target_image, api_key=None):
+    if not source_image or not target_image:
+        return "No images found. Please try again "
+    api_key = api_key or os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/v2/facial-comparison"
+    headers = {
+        'Accept': 'application/json',
+        'Authorization': f'Bearer {api_key}'
+    }
+   
+    if source_image.startswith('http://') or source_image.startswith('https://'):
+        response = requests.get(source_image)
+        source_file=[('source_image', (os.path.basename(source_image), response.content, 'image/jpeg'))]
+    else:
+        with open(source_image, 'rb') as f:
+           source_file=[('source_image', (os.path.basename(source_image), f, 'image/jpeg'))]
+    if target_image.startswith('http://') or target_image.startswith('https://'):
+        response = requests.get(target_image)
+        target_file=[('target_image', (os.path.basename(target_image), response.content, 'image/jpeg'))]
+    else:
+        with open(target_image, 'rb') as f:
+            target_file=[('target_image', (os.path.basename(target_image), f, 'image/jpeg'))]
+    files = source_file + target_file
+    response = requests.post(url, headers=headers, files=files)
+    return response.text

worqhat/ai_models/image_gen.py

@@ -0,0 +1,281 @@
+import requests
+import os
+
+
+def generate_image_v2(prompt="", image_style="realistic", output_type="url", orientation="square", api_key=None):
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+         raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/generate/v2"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "prompt": prompt,
+        "image_style": image_style,
+        "output_type": output_type,
+        "orientation": orientation
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.text
+
+def generate_image_v3(prompt="", image_style="realistic", output_type="url", orientation="square", api_key=None):
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+
+    if not api_key:
+         raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/generate/v3"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "prompt": prompt,
+        "image_style": image_style,
+        "output_type": output_type,
+        "orientation": orientation
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.text
+
+def modify_image_v2(image=None,modification=None, output_type="URL", similarity="30",  api_key=None):
+    if not modification:
+        return "Modification description is missing"
+    if not image or len(image) == 0:
+        return "Image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+         raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/modify/v2"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    payload = {
+            "modification": modification,
+            "output_type": output_type,
+            "similarity": similarity
+        }
+    if image.startswith('http://') or image.startswith('https://'):
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'application/octet-stream'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'application/octet-stream'))]
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+
+def modify_image_v3(image=None,modification=None, output_type="URL", similarity="30",  api_key=None):
+    if not modification:
+        return "Modification description is missing"
+    if not image or len(image) == 0:
+        return "Existing images are missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+
+    # If api_key is still not available, return an error message
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    url = "https://api.worqhat.com/api/ai/images/modify/v3"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+
+
+
+        # Prepare payload
+    payload = {
+            "modification": modification,
+            "output_type": output_type,
+            "similarity": similarity
+        }
+
+        # Prepare files data in the required format
+    if image.startswith('http://') or image.startswith('https://'):
+            # If image is a URL, download the image content
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'application/octet-stream'))]
+    else:
+        # If image is a local file path, read the file content
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'application/octet-stream'))]
+
+        # Make the API call
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+def remove_text_from_image(image=None,output_type="URL",  api_key=None):
+
+    if not image:
+        return "Existing image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/modify/v3/remove-text"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {'output_type': output_type}
+
+    if image.startswith('http://') or image.startswith('https://'):
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+def remove_background_from_image(existing_image=None,output_type="URL",  api_key=None):
+    if not existing_image:
+        return "Existing image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+       raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    url = "https://api.worqhat.com/api/ai/images/modify/v3/remove-background"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    payload = {'output_type': output_type}
+    if existing_image.startswith('http://') or existing_image.startswith('https://'):
+        response = requests.get(existing_image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+        with open(existing_image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+def replace_background(image=None,modification=None, output_type="URL",  api_key=None):
+    if not modification:
+        return "Modification description is missing"
+    if not image:
+        return "Existing image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    url = "https://api.worqhat.com/api/ai/images/modify/v3/replace-background"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {
+        "modification": modification,
+        "output_type": output_type
+    }
+
+    if image.startswith('http://') or image.startswith('https://'):
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+def search_replace_image(image=None,modification=None, output_type="URL", search_object=None,  api_key=None):
+    if not modification:
+        return "Modification description is missing"
+    if not search_object:
+        return "Search object description is missing"
+    if not image:
+        return "Existing image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/modify/v3/search-replace-image"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    payload = {
+        "modification": modification,
+        "output_type": output_type,
+        "search_object": search_object
+    }
+
+    if image.startswith('http://') or image.startswith('https://'):
+
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+  
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+
+def extend_image(image=None,output_type="URL", left_extend="100", right_extend="100", top_extend="50", bottom_extend="50", description="",  api_key=None):
+    if not image:
+        return "image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    url = "https://api.worqhat.com/api/ai/images/modify/v3/extend-image"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+
+    payload = {
+        "output_type": output_type,
+        "leftExtend": left_extend,
+        "rightExtend": right_extend,
+        "topExtend": top_extend,
+        "bottomExtend": bottom_extend,
+        "description": description
+    }
+
+    if image.startswith('http://') or image.startswith('https://'):
+
+        response = requests.get(image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+
+    response = requests.post(url, headers=headers, data=payload, files=files)
+    return response.text
+
+def upscale_image(existing_image=None,scale="4", output_type="URL",api_key=None):
+    if not existing_image:
+        return "Existing image is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+
+    url = "https://api.worqhat.com/api/ai/images/upscale/v3"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    payload = {
+        "scale": scale,
+        "output_type": output_type
+    }
+
+    if existing_image.startswith('http://') or existing_image.startswith('https://'):
+
+        response = requests.get(existing_image)
+        files = [('existing_image', ('file', response.content, 'image/png'))]
+    else:
+        with open(existing_image, 'rb') as f:
+            files = [('existing_image', ('file', f, 'image/png'))]
+
+    response = requests.post(url, headers=headers, data=payload, files=files)
+
+    return response.text

worqhat/ai_models/model_train.py

@@ -0,0 +1,45 @@
+import requests 
+import os
+
+
+def list_datasets(api_key=None):
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/list-datasets"
+    headers = {"Authorization": "Bearer " + api_key}
+    response = requests.post(url, headers=headers)
+    return response.text
+    
+def delete_dataset(dataset_id="", api_key=None):
+    if dataset_id=="":
+        raise ValueError("No Dataset Id provided")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = f"https://api.worqhat.com/api/delete-datasets/{dataset_id}"
+    headers = {"Authorization": "Bearer " + api_key}
+    response = requests.post(url, headers=headers)
+    return response.text
+
+    
+def train_dataset(dataset_id="", dataset_name="", dataset_type="", json_data="", training_file=None, api_key=None):
+    if dataset_id=="":
+        raise ValueError("No Dataset Id provided")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/datasets/train-datasets"
+    headers = {"Authorization": "Bearer " + api_key}
+    payload = {
+            "datasetId": dataset_id,
+            "dataset_name": dataset_name,
+            "dataset_type": dataset_type,
+            "json_data": (None, json_data),
+            "training_file": (training_file, open(training_file, 'rb'))
+    }
+    response = requests.post(url, files=payload, headers=headers)
+    return response.text

worqhat/ai_models/text_extract.py

@@ -0,0 +1,83 @@
+import requests
+import os
+
+
+def extract_pdf_text(pdf_file=None, api_key=None):
+    if not pdf_file:
+        return "PDF file is missing"
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+        print(api_key)
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/v2/pdf-extract"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    files = [('file', (os.path.basename(pdf_file), open(pdf_file, 'rb'), 'application/pdf'))]
+    response = requests.post(url, files=files, headers=headers)
+    return response.text
+
+def web_extract(url_search="", headline=True, inline_code=True, code_blocks=True, references=True, tables=True, api_key=None):
+    if url_search=="":
+        raise ValueError("No URL provided")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/v2/web-extract"
+    headers = {
+        "Authorization": "Bearer " + api_key,
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "url_path": url_search,
+        "headline": headline,
+        "inline_code": inline_code,
+        "code_blocks": code_blocks,
+        "references": references,
+        "tables": tables
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.text
+
+
+def detect_image_text(image=None, output_type="text", api_key=None):
+
+    if not image or len(image) == 0:
+        raise ValueError("Images are missing")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/images/v2/image-text-detection"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    results = []
+    payload = {'output_type': output_type}
+    if image.startswith('http://') or image.startswith('https://'):
+        response = requests.get(image)
+        files = [('image', (os.path.basename(image), response.content, 'image/jpeg'))]
+    else:
+        with open(image, 'rb') as f:
+            files = [('image', (os.path.basename(image), f, 'image/jpeg'))]
+        response = requests.post(url, headers=headers, data=payload, files=files)
+        results.append(response.text)
+    return results
+
+
+def convert_speech_to_text(audio_file=None, api_key=None):
+    if not audio_file:
+        raise ValueError("Audio file is missing")
+    if not api_key:
+        api_key = os.getenv("API_KEY")
+    if not api_key:
+        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
+    url = "https://api.worqhat.com/api/ai/speech-text"
+    headers = {
+        "Authorization": f"Bearer {api_key}"
+    }
+    files = [('audio', (os.path.basename(audio_file), open(audio_file, 'rb'), 'application/octet-stream'))]
+    response = requests.post(url, files=files, headers=headers)
+    return response.text