Merge pull request #152 from yjg30737/Dev

v1.2.0

Author: yjg30737

pyproject.toml

@@ -4,21 +4,25 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pyqt-openai"
-version = "1.1.1"
+version = "1.2.0"
 description = "Python multipurpose chatbot that user can use GPT, other AI models altogether (Release Name: VividNode)"
 authors = [{ name = "Jung Gyu Yoon", email = "[email protected]" }]
 license = { text = "MIT" }
 readme = "README.md"
 dependencies = [
     "PySide6",
-    "openai",
     "pyperclip",
     "jinja2",
-    "llama-index",
     "requests",
-    "replicate"
+    "pyaudio",
+
+    "openai",
+    "anthropic",
+    "google-generativeai",
+    "llama-index",
+    "replicate",
 ]
-keywords = ['openai', 'pyqt', 'pyqt5', 'pyqt6', 'pyside6', 'desktop', 'app', 'chatbot', 'gpt', 'replicate']
+keywords = ['openai', 'pyqt', 'pyqt5', 'pyqt6', 'pyside6', 'desktop', 'app', 'chatbot', 'gpt', 'replicate', 'gemini', 'claude', 'llama', 'llm']
 
 requires-python = ">= 3.11"
 # 3.11

pyqt_openai/__init__.py

@@ -1,5 +1,13 @@
 """
-This file is used to store the constants and the global variables that are used throughout the application.
+This file is used to store the constants and the global constants / variables that are used throughout the application.
+Constants/Variables that are stored here are used throughout the application for the following purposes:
+- Initial values related to the environment settings
+- Values used internally within the application (e.g., application name, DB name, table name, etc.)
+- Values related to the design and UI of the application
+- Default LLM list for the application settings
+
+Constants/Variables that are stored here are not supposed to be changed during the runtime except for __init__.py.
+Variables which are used globally and can be changed are stored in globals.py.
 """
 
 import json
@@ -11,6 +19,7 @@
 from pathlib import Path
 
 # Load the pyproject.toml file
+
 SRC_DIR = Path(__file__).resolve().parent # VividNode/pyqt_openai
 ROOT_DIR = SRC_DIR.parent # VividNode
 SETUP_FILENAME = ROOT_DIR / "pyproject.toml"
@@ -100,6 +109,9 @@ def move_updater():
 HOW_TO_GET_OPENAI_API_KEY_URL = 'https://medium.com/@yjg30737/how-to-get-your-openai-api-key-e2193850932e'
 HOW_TO_EXPORT_CHATGPT_CONVERSATION_HISTORY_URL = 'https://medium.com/@yjg30737/how-to-export-your-chatgpt-conversation-history-caa0946d6349'
 HOW_TO_REPLICATE = 'https://medium.com/@yjg30737/10a2cb983ceb'
+HOW_TO_GET_CLAUDE_API_KEY_URL = 'https://medium.com/@yjg30737/e771e42c182a'
+HOW_TO_GET_GEMINI_API_KEY_URL = 'https://medium.com/@yjg30737/e61a84d64c69'
+HOW_TO_GET_LLAMA_API_KEY_URL = 'https://medium.com/@yjg30737/0dc1900e3606'
 
 COLUMN_TO_EXCLUDE_FROM_SHOW_HIDE_CHAT = ['id']
 COLUMN_TO_EXCLUDE_FROM_SHOW_HIDE_IMAGE = ['id', 'data']
@@ -124,35 +136,15 @@ def move_updater():
 
 MESSAGE_PADDING = 16
 MESSAGE_MAXIMUM_HEIGHT = 800
-MAXIMUM_MESSAGES_IN_PARAMETER = 20
 MESSAGE_MAXIMUM_HEIGHT_RANGE = 300, 1000
-MAXIMUM_MESSAGES_IN_PARAMETER_RANGE = 2, 1000
 
 CONTEXT_DELIMITER = '\n'*2
 PROMPT_IMAGE_SCALE = 200, 200
 TOAST_DURATION = 3
 
-## OPENAI
-OPENAI_REQUEST_URL = 'https://api.openai.com/v1/models'
-
-## PARAMETER - OPENAI CHAT
-OPENAI_TEMPERATURE_RANGE = 0, 2
-OPENAI_TEMPERATURE_STEP = 0.01
-
-MAX_TOKENS_RANGE = 512, 128000
-
-TOP_P_RANGE = 0, 1
-TOP_P_STEP = 0.01
-
-FREQUENCY_PENALTY_RANGE = 0, 2
-FREQUENCY_PENALTY_STEP = 0.01
-
-PRESENCE_PENALTY_RANGE = 0, 2
-PRESENCE_PENALTY_STEP = 0.01
-
+## ICONS
 ICON_PATH = os.path.join(EXEC_PATH, 'ico')
 
-## ICONS
 DEFAULT_APP_ICON = os.path.join(EXEC_PATH, 'icon.ico')
 
 ICON_ADD = os.path.join(ICON_PATH, 'add.svg')
@@ -188,6 +180,8 @@ def move_updater():
 ICON_VERTICAL_THREE_DOTS = os.path.join(ICON_PATH, 'vertical_three_dots.svg')
 ICON_WORD = os.path.join(ICON_PATH, 'word.svg')
 ICON_SEND = os.path.join(ICON_PATH, 'send.svg')
+ICON_RECORD = os.path.join(ICON_PATH, 'record.svg')
+ICON_SPEAKER = os.path.join(ICON_PATH, 'speaker.svg')
 
 ## CUSTOMIZE
 DEFAULT_ICON_SIZE = (24, 24)
@@ -241,6 +235,7 @@ def move_updater():
 DEFAULT_SHORTCUT_LEFT_SIDEBAR_WINDOW = 'Ctrl+L'
 DEFAULT_SHORTCUT_RIGHT_SIDEBAR_WINDOW = 'Ctrl+R'
 DEFAULT_SHORTCUT_CONTROL_PROMPT_WINDOW = 'Ctrl+Shift+C'
+DEFAULT_SHORTCUT_RECORD = 'Ctrl+Shift+R'
 DEFAULT_SHORTCUT_SETTING = 'Ctrl+Alt+S'
 DEFAULT_SHORTCUT_SEND = 'Ctrl+Return'
 
@@ -250,14 +245,6 @@ def move_updater():
 INI_FILE_NAME = os.path.join(get_config_directory(), 'config.yaml')
 LANGUAGE_FILE = os.path.join(get_config_directory(), 'translations.json')
 
-# TODO WILL_REMOVE_AFTER v1.2.0
-prev_lang_file = os.path.join(ROOT_DIR, 'lang/translations.json') if os.path.exists(os.path.join(ROOT_DIR, 'lang/translations.json')) else (os.path.join(SRC_DIR, 'lang/translations.json') if os.path.exists(os.path.join(SRC_DIR, 'lang/translations.json')) else None)
-if prev_lang_file:
-    if os.path.exists(prev_lang_file):
-        shutil.copy(prev_lang_file, LANGUAGE_FILE)
-else:
-    pass
-
 DB_FILE_NAME = 'conv'
 FILE_NAME_LENGTH = 32
 QFILEDIALOG_DEFAULT_DIRECTORY = os.path.expanduser('~')
@@ -317,7 +304,97 @@ def move_updater():
 PROMPT_GROUP_TABLE_NAME = 'prompt_group_tb'
 PROMPT_ENTRY_TABLE_NAME = 'prompt_entry_tb'
 
-# DEFAULT JSON FILENAME FOR PROMPT
+# AI
+## OPENAI
+OPENAI_REQUEST_URL = 'https://api.openai.com/v1/models'
+
+## PARAMETER - OPENAI CHAT
+OPENAI_TEMPERATURE_RANGE = 0, 2
+OPENAI_TEMPERATURE_STEP = 0.01
+
+MAX_TOKENS_RANGE = 512, 128000
+
+TOP_P_RANGE = 0, 1
+TOP_P_STEP = 0.01
+
+FREQUENCY_PENALTY_RANGE = 0, 2
+FREQUENCY_PENALTY_STEP = 0.01
+
+PRESENCE_PENALTY_RANGE = 0, 2
+PRESENCE_PENALTY_STEP = 0.01
+
+## OPENAI WHISPER (TTS, STT)
+WHISPER_TTS_VOICE_TYPE = ['alloy', 'echo', 'fable', 'onyx', 'nova', 'shimmer']
+WHISPER_TTS_VOICE_SPEED_RANGE = 0.25, 4.0
+WHISPER_TTS_MODEL = 'tts-1'
+WHISPER_TTS_DEFAULT_VOICE = 'alloy'
+WHISPER_TTS_DEFAULT_SPEED = 1.0
+
+STT_MODEL = 'whisper-1'
+
+# Endpoint
+# https://platform.openai.com/docs/models/model-endpoint-compatibility
+OPENAI_ENDPOINT_DICT = {
+    '/v1/chat/completions': ['gpt-4o', 'gpt-4o-mini'],
+    '/v1/completions': [
+        'text-davinci-003', 'text-davinci-002', 'text-curie-001', 'text-babbage-001', 'text-ada-001', 'davinci',
+        'curie', 'babbage', 'ada'
+    ],
+    '/v1/edits': ['text-davinci-edit-001', 'code-davinci-edit-001'],
+    '/v1/audio/transcriptions': ['whisper-1'],
+    '/v1/audio/translations': ['whisper-1'],
+    '/v1/fine-tunes': ['davinci', 'curie', 'babbage', 'ada'],
+    '/v1/embeddings': ['text-embedding-ada-002', 'text-search-ada-doc-001'],
+    '/vi/moderations': ['text-moderation-stable', 'text-moderation-latest']
+}
+
+# This doesn't need endpoint
+DALLE_ARR = ['dall-e-2', 'dall-e-3']
+
+OPENAI_CHAT_ENDPOINT = '/v1/chat/completions'
+
+# Other models' configuration data
+DEFAULT_GEMINI_MODEL = 'gemini-1.5-flash'
+LLAMA_REQUEST_URL = "https://api.llama-api.com"
+
+# Overall API configuration data
+DEFAULT_API_CONFIGS = [
+    {
+        "display_name": "OpenAI",
+        "env_var_name": "OPENAI_API_KEY",
+        "api_key": ''
+    },
+    {
+        "display_name": "Gemini",
+        "env_var_name": "GEMINI_API_KEY",
+        "api_key": ''
+    },
+    {
+        "display_name": "Claude",
+        "env_var_name": "CLAUDE_API_KEY",
+        "api_key": ''
+    },
+    {
+        "display_name": "Llama",
+        "env_var_name": "LLAMA_API_KEY",
+        "api_key": ''
+    }
+]
+
+# Dictionary that stores the platform and model pairs
+PLATFORM_MODEL_DICT = {
+    'OpenAI': ['gpt-4o', 'gpt-4o-mini'],
+    'Gemini': ['gemini-1.5-flash'],
+    'Claude': ['claude-3-5-sonnet-20240620'],
+    'Llama': ['llama3-70b']
+}
+
+# Constants related to the number of messages LLM will store
+MAXIMUM_MESSAGES_IN_PARAMETER = 40
+MAXIMUM_MESSAGES_IN_PARAMETER_RANGE = 2, 1000
+
+# PROMPT
+## DEFAULT JSON FILENAME FOR PROMPT
 AWESOME_CHATGPT_PROMPTS_FILENAME = 'prompt_res/awesome_chatgpt_prompts.json'
 ALEX_BROGAN_PROMPT_FILENAME = 'prompt_res/alex_brogan.json'
 
@@ -357,12 +434,45 @@ def move_updater():
     }
 ]'''
 
-# Load the default prompt
+## Load the default prompt
 if os.path.exists(AWESOME_CHATGPT_PROMPTS_FILENAME):
     AWESOME_CHATGPT_PROMPTS = json.load(open(AWESOME_CHATGPT_PROMPTS_FILENAME))[0]
 if os.path.exists(ALEX_BROGAN_PROMPT_FILENAME):
     ALEX_BROGAN_PROMPT = json.load(open(ALEX_BROGAN_PROMPT_FILENAME))[0]
 
+## Data for random prompt generating feature for image generation
+hair_color_randomizer = ['blonde hair', 'red hair', 'blue hair', 'dark hair', 'green hair', 'white hair']
+hair_style_randomizer = ['Long straight hair', 'Short pixie cut', 'Bob haircut', 'Layered haircut', 'Curly hair', 'Wavy hair', 'Ponytail', 'Messy bun', 'French braid', 'Dutch braid', 'Fishtail braid', 'High bun', 'Low bun', 'Top knot', 'Half-up half-down hairstyle', 'Braided crown', 'Side-swept bangs', 'Bangs/fringe', 'Chignon', 'Waterfall braid', 'Mohawk', 'Beach waves', 'Updo', 'French twist', 'Cornrows', 'Dreadlocks', 'Pigtails', 'Space buns', 'Sock bun']
+clothes_randomizer = ['wearing denim jacket', 'wearing coat', 'wearing blazer', 'wearing blouse', 'wearing biker clothes', 'wearing distressed jeans', 'wearing hoodie']
+expression_randomizer = ['Smile', 'Grin', 'Closed eyes', 'Squinting', '(laughing:1.2)', '(giggle:1.2)', '(Angry face:1.4)', 'Frowning face', 'Confused face', 'Boredom', 'Eyebrow furrow', 'Eye roll', 'Smirk']
+pose_randomizer = ['sitting', 'lying', 'leaning', 'cross-legged']
+camera_distance_randomizer = ['(extreme close-up:1.5)', '(medium full shot:1.5)', '(close-up:1.5)', '(establishing shot:1.5)', '(medium close-up:1.5)', '(point-of-view:1.5)', '(medium shot:1.5)', '(cowboy shot:1.5)', '(long shot:1.5)', '(upper body:1.5)', '(full shot:1.5)', '(full body:1.5)']
+camera_angle_randomizer = ['(front view:1.5)', '(from below:1.5)', '(bilaterally symmetrical:1.5)', '(from behind:1.5)', '(side view:1.5)', '(wide angle view:1.5)', '(back view:1.5)', '(fisheyes view:1.5)',
+                           '(from above:1.5)', '(macro view:1.5)',
+                           '(overhead shot:1.5)', '(straight on:1.5)', '(top down view:1.5)', '(hero view:1.5)',
+                           "(bird's eye view:1.5)", '(low view:1.5)', '(high angle:1.5)',
+                           "(worm's eye view:1.5)", '(slightly above:1.5)', '(selfie:1.5)']
+
+adding_lighting_randomizer = ['bloom', 'backlight', 'sun light', 'soft lighting', 'god rays', 'studio light', 'hard lighting', 'volumetic lighting', 'bioluminescent light']
+wind_randomizer = ['(wind:0)', '(wind:0.5)', '(wind:1)', '(wind:1.5)', '(wind:2)']
+acc1_randomizer = ['wearing sunglasses', 'tattoo']
+acc3_randomizer = ['wearing beanie', 'wearing beret', 'wearing cap', 'wearing fedora']
+attr2_randomizer = ['put on earphones', 'piercing on nose', 'put on headphones']
+location_randomizer = ['In the office', 'Tokyo street', 'In the living room', 'In the music studio', 'On the beach', 'In the club']
+
+RANDOMIZING_PROMPT_SOURCE_ARR = [
+    hair_color_randomizer,
+    hair_style_randomizer,
+    clothes_randomizer,
+    expression_randomizer,
+    pose_randomizer,
+    camera_distance_randomizer,
+    camera_angle_randomizer,
+    adding_lighting_randomizer,
+    wind_randomizer,
+    acc1_randomizer,
+]
+
 # DEFAULT Configuration data for the application settings
 # Initialize here to avoid circular import
 # ----------------------------
@@ -399,10 +509,16 @@ def move_updater():
         'ai_image': DEFAULT_AI_IMAGE_PATH,
         'font_size': DEFAULT_FONT_SIZE,
         'font_family': DEFAULT_FONT_FAMILY,
-        'API_KEY': '',
         'llama_index_directory': '',
         'apply_user_defined_styles': False,
         'focus_mode': False,
+        'voice': WHISPER_TTS_DEFAULT_VOICE,
+        'voice_speed': WHISPER_TTS_DEFAULT_SPEED,
+
+        'OPENAI_API_KEY': '',
+        'GEMINI_API_KEY': '',
+        'CLAUDE_API_KEY': '',
+        'LLAMA_API_KEY': ''
     },
     'DALLE': {
         'quality': 'standard',
@@ -441,9 +557,16 @@ def move_updater():
     }
 }
 
+# Dynamically add the API keys to the configuration data
+def update_general_config_with_api_keys(config_data, api_configs):
+    for config in api_configs:
+        config_data['General'][config['env_var_name']] = config['api_key']
+
+update_general_config_with_api_keys(CONFIG_DATA, DEFAULT_API_CONFIGS)
+
 # Update the __all__ list with the PEP8 standard dunder names
 __all__ = ['__version__',
            '__author__']
 
 # Update the __all__ list with the constants
-__all__.extend([name for name, value in globals().items() if name.isupper()])
+__all__.extend([name for name, value in globals().items() if name.isupper()])