commandchat 0.0.4__tar.gz

commandchat-0.0.4/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 cxoto
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

commandchat-0.0.4/PKG-INFO

@@ -0,0 +1,65 @@
+Metadata-Version: 2.1
+Name: commandchat
+Version: 0.0.4
+Summary: use command to chat with openai models
+Home-page: https://github.com/
+Author: xoto
+Author-email: xxx.tao.c@gmail.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+[English](README.md) | [中文](README-zh.md)
+
+# openai-commandchat 
+
+## Quick start
+
+Use this command line to install the python packages:
+
+```bash
+pip3 install commandchat
+```
+Or use the following command line to guide you to a specific python version, if you have multiple versions of python locally
+```bash
+python3.x -m pip3 install commandchat
+```
+
+
+
+## Features
+- You can set the api_key for openai with a simple command, and you can distinguish the key for your multiple accounts with a profile (default `default` if you don't specify it): `occ configure -profile`
+- You can use the command `occ chat` to chat with the chat-GPT model `gpt-3.5-turbo`.
+### To be implemented
+- You can use the `occ image -desc` command with chat-GPT's model to describe the image you want to generate, giving you back a link to the image.
+- You can use the `occ xxx -yyy` command to chat with a specified model of chat-GPT to complete a task you specify or give you advice.
+- Others not yet thought of
+
+
+## Requirements
+
+A terminal with Python and pip installed:
+
+- Installed python environment and python version >= 3.9
+- Install the openai package
+
+## Installing
+
+```bash
+pip3 install commandchat
+```
+
+## Uninstall
+
+```bash
+pip3 uninstall commandchat
+```
+
+## Feedback or questions
+- You can send an email to: ``xxx.tao.c@gmail.com`` or ``xoto@outlook.be``
+
+## License
+MIT
+

commandchat-0.0.4/README.md

@@ -0,0 +1,52 @@
+[English](README.md) | [中文](README-zh.md)
+
+# openai-commandchat 
+
+## Quick start
+
+Use this command line to install the python packages:
+
+```bash
+pip3 install commandchat
+```
+Or use the following command line to guide you to a specific python version, if you have multiple versions of python locally
+```bash
+python3.x -m pip3 install commandchat
+```
+
+
+
+## Features
+- You can set the api_key for openai with a simple command, and you can distinguish the key for your multiple accounts with a profile (default `default` if you don't specify it): `occ configure -profile`
+- You can use the command `occ chat` to chat with the chat-GPT model `gpt-3.5-turbo`.
+### To be implemented
+- You can use the `occ image -desc` command with chat-GPT's model to describe the image you want to generate, giving you back a link to the image.
+- You can use the `occ xxx -yyy` command to chat with a specified model of chat-GPT to complete a task you specify or give you advice.
+- Others not yet thought of
+
+
+## Requirements
+
+A terminal with Python and pip installed:
+
+- Installed python environment and python version >= 3.9
+- Install the openai package
+
+## Installing
+
+```bash
+pip3 install commandchat
+```
+
+## Uninstall
+
+```bash
+pip3 uninstall commandchat
+```
+
+## Feedback or questions
+- You can send an email to: ``xxx.tao.c@gmail.com`` or ``xoto@outlook.be``
+
+## License
+MIT
+

commandchat-0.0.4/commandchat.egg-info/PKG-INFO

@@ -0,0 +1,65 @@
+Metadata-Version: 2.1
+Name: commandchat
+Version: 0.0.4
+Summary: use command to chat with openai models
+Home-page: https://github.com/
+Author: xoto
+Author-email: xxx.tao.c@gmail.com
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Description-Content-Type: text/markdown
+License-File: LICENSE
+
+[English](README.md) | [中文](README-zh.md)
+
+# openai-commandchat 
+
+## Quick start
+
+Use this command line to install the python packages:
+
+```bash
+pip3 install commandchat
+```
+Or use the following command line to guide you to a specific python version, if you have multiple versions of python locally
+```bash
+python3.x -m pip3 install commandchat
+```
+
+
+
+## Features
+- You can set the api_key for openai with a simple command, and you can distinguish the key for your multiple accounts with a profile (default `default` if you don't specify it): `occ configure -profile`
+- You can use the command `occ chat` to chat with the chat-GPT model `gpt-3.5-turbo`.
+### To be implemented
+- You can use the `occ image -desc` command with chat-GPT's model to describe the image you want to generate, giving you back a link to the image.
+- You can use the `occ xxx -yyy` command to chat with a specified model of chat-GPT to complete a task you specify or give you advice.
+- Others not yet thought of
+
+
+## Requirements
+
+A terminal with Python and pip installed:
+
+- Installed python environment and python version >= 3.9
+- Install the openai package
+
+## Installing
+
+```bash
+pip3 install commandchat
+```
+
+## Uninstall
+
+```bash
+pip3 uninstall commandchat
+```
+
+## Feedback or questions
+- You can send an email to: ``xxx.tao.c@gmail.com`` or ``xoto@outlook.be``
+
+## License
+MIT
+

commandchat-0.0.4/commandchat.egg-info/SOURCES.txt

@@ -0,0 +1,20 @@
+LICENSE
+README.md
+setup.py
+commandchat.egg-info/PKG-INFO
+commandchat.egg-info/SOURCES.txt
+commandchat.egg-info/dependency_links.txt
+commandchat.egg-info/entry_points.txt
+commandchat.egg-info/requires.txt
+commandchat.egg-info/top_level.txt
+occ/CommandChat.py
+occ/__init__.py
+occ/command/__init__.py
+occ/command/__main__.py
+occ/commons/__init__.py
+occ/commons/config.py
+occ/configuration/__init__.py
+occ/configuration/profile_config.py
+occ/utils/CommonUtil.py
+occ/utils/__init__.py
+occ/utils/logger.py

commandchat-0.0.4/commandchat.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

commandchat-0.0.4/commandchat.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+occ = occ.command.__main__:main

commandchat-0.0.4/commandchat.egg-info/requires.txt

@@ -0,0 +1,3 @@
+click
+Image
+openai

commandchat-0.0.4/commandchat.egg-info/top_level.txt

@@ -0,0 +1 @@
+occ

commandchat-0.0.4/occ/CommandChat.py

@@ -0,0 +1,116 @@
+import json
+import openai
+import os
+
+from occ.commons.config import get_env
+from occ.utils import logger
+from occ.utils.CommonUtil import save_and_copy_image, waiting_start, waiting_stop
+
+
+def get_home_path():
+    homedir = os.environ.get('HOME', None)
+    if os.name == 'nt':
+        homedir = os.path.expanduser('~')
+    return homedir
+
+
+class CommandChat:
+    DEFAULT_PROFILE = "default"
+    DEFAULT_CHAT_LOG_ID = "chat-1"
+
+    def __init__(self, profile=None, chat_log_id=None):
+        self.api_key = get_env(profile or self.DEFAULT_PROFILE, "api_key")
+        self.api_base = get_env(profile or self.DEFAULT_PROFILE, "api_base_url")
+        self.limit_history = int(get_env(profile or self.DEFAULT_PROFILE, "limit_history") or 4)
+        self.chat_log_id = chat_log_id or self.DEFAULT_CHAT_LOG_ID
+        self.folder_path = os.path.join(get_home_path(), ".occ", profile or self.DEFAULT_PROFILE)
+        self.image_folder_path = os.path.join(self.folder_path, "images")
+        self.file_name = os.path.join(self.folder_path, f"{self.chat_log_id}.log")
+        os.makedirs(self.folder_path, exist_ok=True)
+        os.makedirs(self.image_folder_path, exist_ok=True)
+        if not os.path.exists(self.file_name):
+            open(self.file_name, 'w').close()
+        self.messages = [json.loads(line) for line in (line.strip() for line in open(self.file_name)) if line.strip()]
+
+    def image_create(self, description, size, num):
+        openai.api_key = self.api_key
+        openai.api_base = self.api_base
+        try:
+            response = openai.Image.create(
+                prompt=description,
+                n=num,
+                size=size
+            )
+            for index in range(num):
+                image_url = response['data'][index]['url']
+                save_and_copy_image(image_url, self.image_folder_path)
+        except openai.error.OpenAIError as e:
+            print(e.http_status)
+            print(e.error)
+
+    def image_create_variation(self, img_file, size):
+        openai.api_key = self.api_key
+        openai.api_base = self.api_base
+        try:
+            response = openai.Image.create_variation(
+                open(img_file, "rb"),
+                n=1,
+                size=size
+            )
+            save_and_copy_image(response['data'][0]['url'], self.image_folder_path)
+        except openai.error.OpenAIError as e:
+            print(e.http_status)
+            print(e.error)
+
+    def chat(self, message):
+        openai.api_key = self.api_key
+        openai.api_base = self.api_base
+        message = {"role": "user", "content": message}
+        self.messages.append(message)
+        waiting_start()
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=self.messages,
+            temperature=0.2,
+            top_p=1,
+            frequency_penalty=0.0,
+            presence_penalty=0.5,
+            stream=True
+        )
+        waiting_stop()
+        completion_text = ''
+        content = ''
+        role = None
+        for event in response:
+            if event['choices'][0]["finish_reason"] == "stop":
+                break
+            if role is None:
+                try:
+                    role = event['choices'][0]["delta"]["role"]
+                except Exception:
+                    content = event['choices'][0]["delta"]["content"]
+            else:
+                content = event['choices'][0]["delta"]["content"]
+            completion_text += content
+            print(content, end="")
+        print("\n")
+        self.record_chat_logs(message, {"role": role, "content": completion_text.replace("\n\n", "")})
+
+    def record_chat_logs(self, content, completion_text):
+        with open(self.file_name, 'r+') as f:
+            lines = f.readlines()
+            if len(lines) >= self.limit_history:
+                limit_history_ = (len(lines) + 2 - self.limit_history)
+                with open(os.path.join(self.folder_path, self.chat_log_id + '_history.log'), 'a+') as hf:
+                    hf.writelines("\n")
+                    hf.writelines(lines[:limit_history_])
+                lines = lines[limit_history_:]
+            if len(lines) == 0:
+                lines.append('{}\n{}'.format(json.dumps(content, ensure_ascii=False),
+                                             json.dumps(completion_text, ensure_ascii=False)))
+            else:
+                lines.append('\n{}\n{}'.format(json.dumps(content, ensure_ascii=False),
+                                               json.dumps(completion_text, ensure_ascii=False)))
+            f.seek(0)
+            f.truncate()
+            f.writelines(lines)

commandchat-0.0.4/occ/command/__main__.py

@@ -0,0 +1,70 @@
+from __future__ import absolute_import
+
+import click
+import pkg_resources
+
+import occ.utils.logger as logger
+from occ.configuration.profile_config import add_profile, add_default_profile
+from occ.CommandChat import CommandChat
+from occ.utils.CommonUtil import waiting_stop
+
+VERSION = pkg_resources.require("commandchat")[0].version
+
+
+@click.group()
+@click.version_option(version=VERSION, prog_name='openai-commandchat')
+def commandchat_operator():
+    pass
+
+
+@click.command()
+@click.option('-profile', help='Enable profile name')
+def configure(profile):
+    if profile is not None:
+        add_profile(profile)
+    else:
+        add_default_profile()
+
+
+@click.command()
+@click.argument('message')
+@click.option('-id', help=' enter chat id, something like context')
+@click.option('-profile', help='Enable profile name')
+def chat(message, id, profile):
+    try:
+        CommandChat(profile=profile, chat_log_id=id).chat(message)
+    except Exception as e:
+        logger.log_g(str(e))
+        waiting_stop()
+
+
+size_map = {
+    "s": "256x256",
+    "S": "256x256",
+    "m": "512x512",
+    "M": "512x512",
+    "l": "1024x1024",
+    "L": "1024x1024"
+}
+
+
+@click.command()
+@click.option('-desc', help=' Enter the description of the images you want')
+@click.option('-size',
+              help=' Enter the size(S/s,M/m,L/l): \n   small - 256x256 \n   middle  - 512x512 \n   large - 1024x1024')
+@click.option('-num', count=True, help=' Enter the number to generate the specified number of images')
+@click.option('-profile', help='Enable profile name')
+def image(desc, size, num, profile):
+    number = num if num > 0 else 1
+    size = size_map.get(size)
+    size_value = size if size is not None else "512x512"
+    CommandChat(profile=profile).image_create(desc, size_value, number if number < 5 else 4)
+
+
+commandchat_operator.add_command(configure)
+commandchat_operator.add_command(chat)
+commandchat_operator.add_command(image)
+
+
+def main():
+    commandchat_operator()

commandchat-0.0.4/occ/commons/config.py

@@ -0,0 +1,99 @@
+import configparser
+import logging
+import os
+
+import pkg_resources
+
+import occ.utils.logger as logger
+
+config = configparser.SafeConfigParser()
+
+
+def get_home_path():
+    homedir = os.environ.get('HOME', None)
+    if os.name == 'nt':
+        homedir = os.path.expanduser('~')
+    return homedir
+
+
+def get_config_path():
+    homedir = get_home_path()
+    if not homedir:
+        logger.log_r("Home Directory Not found!! Set Envirnoment `HOME` ")
+        exit()
+    logger.debug("Home Directory : " + homedir)
+    config_file_temp = os.path.join(homedir + "/.occ/config")
+    logger.debug("Config File Location : " + config_file_temp)
+    if not os.path.exists(config_file_temp):
+        logger.log_r("ERROR: No Config file present")
+        try:
+            os.makedirs(os.path.dirname(config_file_temp))
+        except OSError as exc:  # Guard against race condition
+            logger.log_r("Directory found! but not config file")
+
+        logger.debug("Creating config file")
+        file = open(config_file_temp, "w")
+        file.write("[default]")
+        file.close()
+
+    logger.debug(config.read(config_file_temp))
+    return config_file_temp
+
+
+config_file = get_config_path()
+
+
+def set_env(profile, key, value):
+    config.set(profile, key, value)
+    write_config()
+
+
+def add_profile(profile):
+    if config.has_section(profile):
+        logger.log_r(profile + " Section already exists!!")
+        return
+
+    config.add_section(profile)
+    write_config()
+
+
+def write_config():
+    with open(config_file, 'w') as configfile:
+        config.write(configfile)
+
+
+def get_env(profile, key):
+    logger.debug("Searching in profile : " + profile)
+    logger.debug("Searching in key" + key)
+    if config.has_option(profile, key):
+        return config.get(profile, key)
+    logger.debug("Not found in current profile")
+    if config.has_option('default', key):
+        return config.get('default', key)
+
+    logger.debug("No Value Found in DEAFULT SECTION as well")
+    logger.log_r(
+        'Value not found in [Default Profile] use `occ configure`comamnd')
+    exit()
+
+
+def get_default_env(key):
+    if config.has_option('default', key):
+        return config.get('default', key)
+    return None
+
+
+def get_profiles():
+    return config.sections()
+
+
+def log_config():
+    level_input = 'DEBUG'
+    if level_input == 'DEBUG':
+        print(level_input)
+        logging.basicConfig(level=logging.DEBUG)
+
+
+def find_version():
+    version = pkg_resources.require("recordset")[0].version
+    print(version)

commandchat-0.0.4/occ/configuration/profile_config.py

@@ -0,0 +1,36 @@
+import click
+
+import occ.commons.config as config
+import occ.utils.logger as logger
+
+
+def add_default_profile():
+    api_key = click.prompt(logger.style('\n[default] Your default api_key '),
+                           default=config.get_default_env("api_key"), type=str)
+    config.set_env('default', 'api_key', api_key)
+
+    api_key = click.prompt(logger.style('\n[default] Your api_base_url(Optional) '),
+                           default=config.get_default_env("api_base_url"), type=str)
+    config.set_env('default', 'api_base_url', api_key)
+
+    limit_history = click.prompt(logger.style('\n[default] Your default limit_history '),
+                                 default=config.get_default_env("limit_history"), type=str)
+    config.set_env('default', 'limit_history', limit_history)
+
+
+def input_config_vars(profile_name, key_name, is_default_exist):
+    if is_default_exist:
+        default_value = config.get_env('default', key_name)
+        user_input = click.prompt(logger.style('\ndefault value  ' + key_name), default=default_value, type=str)
+    else:
+        user_input = click.prompt(logger.style('\nYour ' + key_name), type=str)
+
+    config.set_env(profile_name, key_name, user_input)
+
+
+def add_profile(profile_name):
+    config.add_profile(profile_name)
+    config.set_env(profile_name, 'profile_name', profile_name)
+    input_config_vars(profile_name, 'api_key', False)
+    input_config_vars(profile_name, 'api_base_url', False)
+    input_config_vars(profile_name, 'limit_history', False)

commandchat-0.0.4/occ/utils/CommonUtil.py

@@ -0,0 +1,53 @@
+import os
+import requests
+from urllib.parse import urlparse
+from PIL import Image
+
+import time
+import sys
+import threading
+
+
+class WaitingThread(threading.Thread):
+    def __init__(self):
+        super().__init__()
+        self._stop_event = threading.Event()
+
+    def run(self):
+        while not self._stop_event.is_set():
+            for i in range(4):
+                sys.stdout.write('\r' + ' ' * 60 + '\r')  # 清空当前行
+                sys.stdout.write('Waiting' + '.' * i + ' ' * (3 - i))
+                sys.stdout.flush()
+                time.sleep(0.5)
+
+    def stop(self):
+        self._stop_event.set()
+
+
+def save_and_copy_image(url, image_path):
+    response = requests.get(url)
+    image_file = os.path.join(image_path, os.path.basename(urlparse(url).path))
+    with open(image_file, 'wb') as f:
+        f.write(response.content)
+    print("image download path: " + image_file)
+    show_image(image_file)
+
+
+def show_image(image):
+    img = Image.open(image)
+    img.show()
+
+
+waiting_thread = WaitingThread()
+
+
+def waiting_start():
+    waiting_thread.start()
+
+
+def waiting_stop():
+    waiting_thread.stop()
+    waiting_thread.join()
+    sys.stdout.write('\r' + ' ' * 60 + '\r')
+    sys.stdout.flush()

commandchat-0.0.4/occ/utils/logger.py

@@ -0,0 +1,72 @@
+import click
+
+debug_logging = False;
+
+all_colors = 'black', 'red', 'green', 'yellow', 'blue', 'magenta', \
+    'cyan', 'white', 'bright_black', 'bright_red', \
+    'bright_green', 'bright_yellow', 'bright_blue', \
+    'bright_magenta', 'bright_cyan', 'bright_white'
+
+
+def log_c(*argv):
+    if len(argv) == 1:
+        click.echo(click.style(argv[0], fg='magenta', bold=True))
+    if len(argv) == 2:
+        click.secho(argv[0], nl=False)
+        click.echo(click.style(argv[1], fg='green', bold=True))
+
+
+def log_r(str2):
+    click.echo(click.style(str2, fg='red', bold=True))
+
+
+def style(str):
+    return click.style(str, fg='cyan', bold=True)
+
+
+def log_cy(str):
+    click.echo(click.style(str, fg='cyan', bold=True))
+
+
+def log_cyb(str):
+    click.echo(click.style(str, fg='cyan', bold=True, blink=True))
+
+
+def log_g(str2):
+    click.echo(click.style(str2, fg='green', bold=True))
+
+
+def log_y(str2):
+    click.echo(click.style(str2, fg='yellow', bold=True))
+
+
+def log_bl(str2):
+    click.echo(click.style(str2, fg='blue', bold=True))
+
+
+# Colored logging
+def debug(str):
+    if debug_logging:
+        click.secho((str))
+
+
+# Colored logging
+def debug_c(str):
+    if debug_logging:
+        click.echo(click.style(str, fg='blue', bold=True))
+
+
+# Colored logging
+def enable_debug():
+    global debug_logging
+    debug_logging = True
+
+
+# Simple logging    
+def log(str):
+    click.echo(click.style(str, fg='green', bold=True))
+
+
+# blinking logging    
+def log_b(str):
+    click.echo(click.style(str, blink=True))

commandchat-0.0.4/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

commandchat-0.0.4/setup.py

@@ -0,0 +1,31 @@
+import setuptools
+
+with open("README.md", "r", encoding='utf-8') as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name='commandchat',
+    version='0.0.4',
+    entry_points={
+        'console_scripts': [
+            'occ = occ.command.__main__:main'
+        ]
+    },
+    author="xoto",
+    author_email="xxx.tao.c@gmail.com",
+    description="use command to chat with openai models",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/",
+    packages=setuptools.find_packages(),
+    install_requires=[
+        'click',
+        'Image',
+        'openai'
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+)