worqhat 0.0.1__py3-none-any.whl → 0.1.0a3__py3-none-any.whl

worqhat/ai_models/image_gen.py

@@ -1,281 +0,0 @@
-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

@@ -1,45 +0,0 @@
-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

@@ -1,83 +0,0 @@
-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

worqhat/ai_models/text_gen.py

@@ -1,177 +0,0 @@
-import requests 
-import json
-import os
-def get_ai_response_v2(question="", 
-                       preserve_history=False, 
-                       randomness=0.5, 
-                       stream_data=False, 
-                       conversation_history=[], 
-                       training_data="", 
-                       response_type="text",
-                       api_key=None):
-    url = "https://api.worqhat.com/api/ai/content/v2"
-
-    payload = {
-        "question": question,
-        "preserve_history": preserve_history,
-        "randomness": randomness,
-        "stream_data": stream_data,
-        "conversation_history": conversation_history,
-        "training_data": training_data,
-        "response_type": response_type
-    }
-
-    headers = {"Content-Type": "application/json"}
-
-    if not api_key:
-        api_key = os.getenv("API_KEY")
-
-    if api_key:
-        headers["Authorization"] = "Bearer " + api_key
-    else:
-        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
-
-    response = requests.post(url, json=payload, headers=headers, stream=stream_data)
-
-    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)
-            return(json.dumps(json_data, indent=4))
-        except json.JSONDecodeError:
-            print(json_response)
-
-
-def get_ai_response_v3(question="", 
-                       preserve_history=False, 
-                       randomness=0.5, 
-                       stream_data=False, 
-                       conversation_history=[], 
-                       training_data="", 
-                       response_type="text",
-                       api_key=None):
-    url = "https://api.worqhat.com/api/ai/content/v3"
-
-    payload = {
-        "question": question,
-        "preserve_history": preserve_history,
-        "randomness": randomness,
-        "stream_data": stream_data,
-        "conversation_history": conversation_history,
-        "training_data": training_data,
-        "response_type": response_type
-    }
-
-    headers = {"Content-Type": "application/json"}
-
-    if not api_key:
-        api_key = os.getenv("API_KEY")
-
-    if api_key:
-        headers["Authorization"] = "Bearer " + api_key
-    else:
-        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
-
-    response = requests.post(url, json=payload, headers=headers, stream=stream_data)
-
-    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 get_alpha_ai_response(question="", 
-                          preserve_history=False, 
-                          randomness=0.5, 
-                          stream_data=False, 
-                          conversation_history=[], 
-                          training_data="", 
-                          response_type="text",
-                          api_key=None):
-    url = "https://api.worqhat.com/api/ai/content/v3/alpha"
-    payload = {
-        "question": question,
-        "preserve_history": preserve_history,
-        "randomness": randomness,
-        "stream_data": stream_data,
-        "conversation_history": conversation_history,
-        "training_data": training_data,
-        "response_type": response_type
-    }
-    headers = {"Content-Type": "application/json"}
-    if not api_key:
-        api_key = os.getenv("API_KEY")
-    if api_key:
-        headers["Authorization"] = "Bearer " + api_key
-    else:
-        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
-    response = requests.request("POST",url, json=payload, headers=headers)
-    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 get_large_ai_response_v2(question="", 
-                             dataset_id="", 
-                             preserve_history=True, 
-                             randomness=0.5, 
-                             stream_data=False, 
-                             conversation_history=[], 
-                             instructions=None,
-                             api_key=None):
-    url = "https://api.worqhat.com/api/ai/content/v2-large/answering"
-    payload = {
-        "question": question,
-        "datasetId": dataset_id,
-        "preserve_history": preserve_history,
-        "randomness": randomness,
-        "stream_data": stream_data,
-        "conversation_history": conversation_history,
-        "instructions": instructions
-    }
-    headers = {"Content-Type": "application/json"}
-    if not api_key:
-        api_key = os.getenv("API_KEY")
-    if api_key:
-        headers["Authorization"] = "Bearer " + api_key
-    else:
-        raise ValueError("API key is missing. Provide it as an argument or in the .env file.")
-    response = requests.request("POST",url, json=payload, headers=headers)
-    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)