dazed/DazedTL
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Push latest

Browse source
This commit is contained in:
DazedAnon 2024-09-13 19:24:44 -05:00
parent 921ea31a4e
commit 0533ca7078
4 changed files with 97 additions and 314 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/images.py
View file

@ -192,7 +192,7 @@ def getImageDimensions(file_path):
def processImagesDir(directory_path, imageList):
for file_name in os.listdir(directory_path):
# .png and Japanese
if '.png' in file_name and re.search(r'[一-龠ぁ-ゔァ-ヴー]+', file_name) and file_name.replace('.png', '') in VOCAB and len(file_name) > 5:
if '.png' in file_name and re.search(r'[一-龠ぁ-ゔァ-ヴー]+', file_name) and file_name.replace('.png', '') in VOCAB:
file_path = os.path.join(directory_path, file_name)
if os.path.isfile(file_path):
# Check if the file is an image

121
modules/wolf.py
View file

@ -49,7 +49,7 @@ if 'gpt-3.5' in MODEL:
elif 'gpt-4' in MODEL:
INPUTAPICOST = .005
OUTPUTAPICOST = .015
BATCHSIZE = 40
BATCHSIZE = 20
FREQUENCY_PENALTY = 0.1
#tqdm Globals
@ -62,12 +62,12 @@ FILENAME = None
# Dialogue / Scroll
CODE101 = False
CODE102 = False
CODE122 = False
CODE122 = True
# Other
CODE210 = False
CODE300 = False
CODE250 = True
CODE250 = False
# Database
NPCFLAG = False
@ -367,34 +367,9 @@ def searchCodes(events, pbar, jobList, filename):
# # Validate and Set Data
# codeList[i]['stringArgs'][1] = translatedText
if 'stringArgs' in codeList[i] and len(codeList[i]['stringArgs']) > 1:
imageRegex = r'(\r?\n?_\w+\r?\n?)|(@-?\d?\r\n)|(@-?\d?[^\r])(\d?-?\d?[^\r]+?)(\r\n|$)|(_PDC)|(>\r\n)|(^[#])|(\r\n@$)|(/)|(_SS_)|(\r\n[@/]-?\d?\r\n)'
cleanedList = formatDramon(codeList[i]['stringArgs'][1])
fontSize = 24
# Grab String
jaString = codeList[i]['stringArgs'][1]
jaString = jaString.replace('\u3000', ' ')
jaString = jaString.replace('#', '')
# Grab and Split
jaStringList = re.split(imageRegex, jaString)
# Clean List
cleanedList = [x for x in jaStringList if x is not None and x != '' and x != '\r\n' and x != '_SS_']
# Iterate Through List
j = 0
translatedText = ''
while j < len(cleanedList):
if ('@' in cleanedList[j] or '/' in cleanedList[j]) and j < len(cleanedList)-1 and re.search(r'([@/]-?\d?\r\n)', cleanedList[j]) is None:
# Setup @
if j > 0 and '@' not in cleanedList[j-1] and '/' not in cleanedList[j-1] and '_' not in cleanedList[j-1]:
cleanedList[j-1] = cleanedList[j-1] + cleanedList[j+1]
else:
cleanedList.insert(j, cleanedList[j+1])
j += 1
cleanedList[j] = f'\r\n{cleanedList[j]}\r\n'
cleanedList.pop(j+1)
j += 1
for str in cleanedList:
# Pass 1
@ -493,7 +468,7 @@ def searchCodes(events, pbar, jobList, filename):
codeList[i]['stringArgs'][0] = translatedText
### Event Code: 300 Common Events
if codeList[i]['code'] == 122 and CODE300 == True and len(codeList[i]['stringArgs']) > 0:
if codeList[i]['code'] == 300 and CODE300 == True and len(codeList[i]['stringArgs']) > 0:
# Choices
if codeList[i]['stringArgs'][0] == "m_可変式選択肢":
# Grab String
@ -531,38 +506,12 @@ def searchCodes(events, pbar, jobList, filename):
or codeList[i]['stringArgs'][0] == 'm_謎冒頭' \
or (codeList[i]['codeStr'] == 'SetString' and '@' in codeList[i]['stringArgs'][0])\
or codeList[i]['stringArgs'][0] == '[移]サウンドノベル':
imageRegex = r'(\r?\n?_\w+\r?\n?)|(@-?\d?\r\n)|(@-?\d?[^\r*])(\d?-?\d?[^\r]+?)(\r\n|$)|(_PDC)|(>\r\n)|(^[#])|(\r\n@$)|(/)|(_SS_)|(\r\n[@/]-?\d?\r\n)'
fontSize = 24
# Grab String
if codeList[i]['codeStr'] != 'SetString':
jaString = codeList[i]['stringArgs'][1]
if len(codeList[i]['stringArgs']) > 1:
cleanedList = formatDramon(codeList[i]['stringArgs'][1])
else:
jaString = codeList[i]['stringArgs'][0]
jaString = jaString.replace('\u3000', ' ')
jaString = jaString.replace('#', '')
jaString = re.sub(r'[^\r]\n', '\r\n', jaString)
# Grab and Split
jaStringList = re.split(imageRegex, jaString)
# Clean List
cleanedList = [x for x in jaStringList if x is not None and x != '' and x != '\r\n' and x != '_SS_']
# Iterate Through List
j = 0
cleanedList = formatDramon(codeList[i]['stringArgs'][0])
fontSize = 24
translatedText = ''
while j < len(cleanedList):
if ('@' in cleanedList[j] or '/' in cleanedList[j]) and j < len(cleanedList)-1 and re.search(r'([@/]-?\d?\r\n)', cleanedList[j]) is None:
# Setup @
if j > 0 and '@' not in cleanedList[j-1] and '/' not in cleanedList[j-1] and '_' not in cleanedList[j-1]:
cleanedList[j-1] = cleanedList[j-1] + cleanedList[j+1]
else:
cleanedList.insert(j, cleanedList[j+1])
j += 1
cleanedList[j] = f'\r\n{cleanedList[j]}\r\n'
cleanedList.pop(j+1)
j += 1
for str in cleanedList:
# Pass 1
@ -597,7 +546,7 @@ def searchCodes(events, pbar, jobList, filename):
translatedText = translatedText.replace('*"', '* "')
translatedText = translatedText.replace('\r\n\r\n', '\r\n')
translatedText = re.sub(r'[^\S\r\n]+', ' ', translatedText)
if codeList[i]['codeStr'] != 'SetString':
if len(codeList[i]['stringArgs']) > 1:
codeList[i]['stringArgs'][1] = translatedText
else:
codeList[i]['stringArgs'][0] = translatedText
@ -694,6 +643,36 @@ def searchCodes(events, pbar, jobList, filename):
return totalTokens
def formatDramon(jaString):
imageRegex = r'(\r?\n?_\w+\r?\n?)|(@-?\d?\r\n)|(@-?\d?)([^\r]\d?-?\d?[^\r]+?)(\r\n|$)|(_PDC)|(>\r\n)|(^[#])|(\r\n@$)|(/)|(_SS_)|(\r\n[@/]-?\d?\r\n)'
jaString = jaString.replace('\u3000', ' ')
jaString = jaString.replace('#', '')
jaString = re.sub(r'[^\r]\n', '\r\n', jaString)
# Grab and Split
jaStringList = re.split(imageRegex, jaString)
# Clean List
cleanedList = [x for x in jaStringList if x is not None and x != '' and x != '\r\n' and x != '_SS_']
# Iterate Through List
j = 0
translatedText = ''
while j < len(cleanedList):
if ('@' in cleanedList[j] or '/' in cleanedList[j]) and j < len(cleanedList)-1 and re.search(r'([@/]-?\d?\r\n)', cleanedList[j]) is None:
# Setup @
if j > 0 and '@' not in cleanedList[j-1] and '/' not in cleanedList[j-1] and '_' not in cleanedList[j-1]:
cleanedList[j-1] = cleanedList[j-1] + cleanedList[j+1]
else:
cleanedList.insert(j, cleanedList[j+1])
j += 1
cleanedList[j] = f'\r\n{cleanedList[j]}\r\n'
cleanedList.pop(j+1)
j += 1
return cleanedList
# DatabaseDatabase
def searchDB(events, pbar, jobList, filename):
# Set Lists
@ -841,7 +820,7 @@ def searchDB(events, pbar, jobList, filename):
scenarioList[2].pop(0)
# Grab Items
if table['name'] == '宿泊施設' and ITEMFLAG == True:
if table['name'] == '商品' and ITEMFLAG == True:
with open('translations.txt', 'a') as file:
for item in table['data']:
dataList = item['data']
@ -849,7 +828,7 @@ def searchDB(events, pbar, jobList, filename):
# Parse #
for j in range(len(dataList)):
# Name
if 'NULL' in dataList[j].get('name'):
if '商品の名前' in dataList[j].get('name'):
# Pass 1 (Grab Data)
if setData == False:
if dataList[j].get('value') != '':
@ -863,7 +842,7 @@ def searchDB(events, pbar, jobList, filename):
itemList[0].pop(0)
# Description
if dataList[j].get('name') == '選2文字列(夜時非表示':
if dataList[j].get('name') == 'NULL':
# Pass 1 (Grab Data)
if setData == False:
if dataList[j].get('value') != '':
@ -883,14 +862,14 @@ def searchDB(events, pbar, jobList, filename):
translatedText = itemList[1][0]
translatedText = textwrap.fill(translatedText, LISTWIDTH)
translatedText = translatedText.replace('\n', '\r\n')
translatedText = font + translatedText
translatedText = font + translatedText.replace('\r\n', f'\r\n{font}')
# Set Data
dataList[j].update({'value': translatedText})
itemList[1].pop(0)
# Description 2
if dataList[j].get('name') == '選3文字列':
if dataList[j].get('name') == 'NULL':
# Pass 1 (Grab Data)
if setData == False:
if dataList[j].get('value') != '':
@ -909,14 +888,15 @@ def searchDB(events, pbar, jobList, filename):
# Textwrap
translatedText = itemList[2][0]
translatedText = textwrap.fill(translatedText, LISTWIDTH)
translatedText = font + translatedText
translatedText = translatedText.replace('\n', '\r\n')
translatedText = font + translatedText.replace('\r\n', f'\r\n{font}')
# Set Data
dataList[j].update({'value': translatedText})
itemList[2].pop(0)
# Description 3
if dataList[j].get('name') == '┗セ⑥_昼(になった)':
if dataList[j].get('name') == 'NULL':
# Pass 1 (Grab Data)
if setData == False:
if dataList[j].get('value') != '':
@ -935,7 +915,8 @@ def searchDB(events, pbar, jobList, filename):
# Textwrap
translatedText = itemList[3][0]
translatedText = textwrap.fill(translatedText, LISTWIDTH)
translatedText = font + translatedText
translatedText = translatedText.replace('\n', '\r\n')
translatedText = font + translatedText.replace('\r\n', f'\r\n{font}')
# Set Data
dataList[j].update({'value': translatedText})
@ -1751,7 +1732,7 @@ def translateGPT(text, history, fullPromptFlag):
continue
# Translating
response = translateText(characters, system, user, history, 0.05, format)
response = translateText(characters, system, user, history, 0.1, format)
translatedText = response.choices[0].message.content
totalTokens[0] += response.usage.prompt_tokens
totalTokens[1] += response.usage.completion_tokens

52
modules/wolf2.py
View file

@ -352,14 +352,50 @@ def batchList(input_list, batch_size):
def createContext(fullPromptFlag, subbedT):
characters = 'Game Characters:\n\
千佳 (Chika) - Female\n\
ちか (Chika) - Female\n\
和樹 (Kazuki) - Male\n\
かずき (Kazuki) - Male\n\
猿山 (Saruyama) - Male\n\
菊池 (Kikuchi) - Male\n\
篠宮 (Shinomiya) - Male\n\
翔太 (Shota) - Male\n\
ロラン (Roland) - Male\n\
リュカ (Ryuka) - Male\n\
レックス (Rex) - Male\n\
タバサ (Tabasa) - Female\n\
アルス (Ars) - Male\n\
アマカラ (Amakara) - Male\n\
エリー (Eri) - Female\n\
リオ (Rio) - Female\n\
サマル (Samal) - Male\n\
ムーン (Moon) - Female\n\
アリーナ (Arina) - Female\n\
クリフト (Cliff) - Male\n\
マーニャ (Manya) - Female\n\
ミネア (Minea) - Female\n\
デボラ (Debora) - Female\n\
ビアンカ (Bianca) - Female\n\
フローラ (Flora) - Female\n\
バーバラ (Barbara) - Female\n\
ミレーユ (Mireyu) - Female\n\
アイラ (Aira) - Female\n\
フォズ (Foz) - Female\n\
マリベル (Maribel) - Female\n\
ククール (Kukool) - Male\n\
ゲルダ (Gerda) - Female\n\
ゼシカ (Jessica) - Female\n\
ヤンガス (Yangus) - Male\n\
ラヴィエル (Raviel) - Female\n\
セティア (Setia) - Female\n\
ダイ (Dai) - Male\n\
ヒュンケル (Hyunckel) - Male\n\
ポップ (Pop) - Male\n\
マァム (Maam) - Female\n\
レオナ (Leona) - Female\n\
アステア (Astea) - Female\n\
イヨ (Iyo) - Female\n\
ジャガン (Jagan) - Male\n\
ヤオ (Yao) - Female\n\
デイジィ (Daisy) - Female\n\
バイシュン (Baishun) - Male\n\
ブライ (Buraimu) - Male\n\
ハッサン (Hassan) - Male\n\
アロマ (Aroma) - Female\n\
ピッケ (Pikke) - Female\n\
ドラオ (Dorao) - Male\n\
'
system = PROMPT + VOCAB if fullPromptFlag else \

236
vocab.txt
View file

@ -67,240 +67,6 @@ ME 音量 (ME Volume)
リーロランド (Liloland)
ローバー (Roper)
# Armors
アマカラ靴1 (Amakara Shoes 1)
お鍋のフタ (Pot Lid)
ライトシールド (Light Shield)
キトンシールド (Kiton Shield)
こうらの盾 (Shell Shield)
プラチナシールド (Platinum Shield)
魔法の盾 (Magic Shield)
シルバートレイ (Silver Tray)
皮の盾 (Leather Shield)
青銅の盾 (Bronze Shield)
ホワイトシールド (White Shield)
力の盾 (Power Shield)
水鏡の盾 (Water Mirror Shield)
ダークシールド (Dark Shield)
精霊の盾 (Spirit Shield)
スチヌラの盾 (Stinura Shield)
輪廻の盾 (Reincarnation Shield)
はぐれスチルの盾 (Stray Steel Shield)
破滅の盾 (Shield of Ruin)
鉄の盾 (Iron Shield)
鋼の盾 (Steel Shield)
氷の盾 (Ice Shield)
炎の盾 (Flame Shield)
玉鋼の盾 (Jewel Steel Shield)
ドラゴンシールド (Dragon Shield)
風神の盾 (Wind God Shield)
オーガシールド (Ogre Shield)
幻魔の盾 (Phantom Demon Shield)
女神の盾 (Goddess Shield)
シャハルの鏡 (Shahar's Mirror)
エンデの盾 (Ende's Shield)
スチルキングの盾 (Steel King Shield)
天空の盾 (Sky Shield)
ロトの盾 (Roto's Shield)
光の盾 (Light Shield)
皮の鎧 (Leather Armor)
くさりかたびら (Chain Mail)
鉄の胸当て (Iron Breastplate)
鋼の胸当て (Steel Breastplate)
プラチナメイル (Platinum Mail)
こうらの鎧 (Shell Armor)
ドラゴンスケイル (Dragon Scale)
シルバーメイル (Silver Mail)
ダンシングメイル (Dancing Mail)
魔法の鎧 (Magic Armor)
精霊の鎧 (Spirit Armor)
神秘の鎧 (Mystic Armor)
スチヌラ鎧 (Stenura Armor)
エンデの鎧 (Ende's Armor)
はぐれスチル鎧 (Stray Steel Armor)
勝利の鎧 (Victory Armor)
鉄の鎧 (Iron Armor)
鋼の鎧 (Steel Armor)
戦士の鎧 (Warrior's Armor)
ドラゴンメイル (Dragon Mail)
大地の鎧 (Earth Armor)
玉鋼の鎧 (Jewel Steel Armor)
氷の鎧 (Ice Armor)
炎の鎧 (Flame Armor)
魔神の鎧 (Demon Armor)
ミスリルアーマー (Mithril Armor)
ギガントアーマー (Gigant Armor)
幻魔の鎧 (Phantom Demon Armor)
女神の鎧 (Goddess Armor)
悪魔の鎧 (Devil Armor)
スチルキング鎧 (Steel King Armor)
布の服 (Cloth Garb)
旅人の服 (Traveler's Clothes)
貴族の服 (Noble's Clothes)
毛皮のコート (Fur Coat)
みかわしの服 (Evasion Clothes)
拳法着 (Martial Arts Wear)
かくれみの服 (Cloak of Invisibility)
竜の道着 (Dragon's Gi)
闇の衣 (Dark Robe)
レザーマント (Leather Mantle)
紅蓮のローブ (Crimson Robe)
やすらぎのローブ (Tranquil Robe)
魔法の法衣 (Magic Vestment)
神官のエプロン (Priest's Apron)
賢者のローブ (Sage's Robe)
天使のローブ (Angel's Robe)
不思議なボレロ (Mysterious Bolero)
精霊の法衣 (Spirit's Garb)
神秘のローブ (Mystic Robe)
ドラゴンローブ (Dragon Robe)
皮のドレス (Leather Dress)
絹のローブ (Silk Robe)
マジカルスカート (Magical Skirt)
シルクのビスチェ (Silk Bustier)
水の羽衣 (Water Robe)
光のドレス (Dress of Light)
プリンセスローブ (Princess Robe)
踊り子の服 (Dancer's Outfit)
バニースーツ (Bunny Suit)
あぶない水着 (Dangerous Swimsuit)
天使のレオタード (Angel's Leotard)
魔法のビキニ (Magic Bikini)
神秘のビスチェ (Mystic Bustier)
清き衣 (Pure Garment)
エッチな下着 (Sexy Underwear)
天空の鎧 (Sky Armor)
王者のマント (King's Mantle)
ロトの鎧 (Roto's Armor)
皮の帽子 (Leather Hat)
サークレット (Circlet)
毛皮のフード (Fur Hood)
神官の帽子 (Priest's Hat)
知識の帽子 (Hat of Knowledge)
ファントムマスク (Phantom Mask)
マジカルハット (Magical Hat)
般若の面 (Hannya Mask)
幸せの帽子 (Hat of Happiness)
不思議な帽子 (Mysterious Hat)
山彦の帽子 (Echo Hat)
うさみみバンド (Bunny Ears Band)
銀の髪飾り (Silver Hair Ornament)
ウィッチハット (Witch Hat)
黄金のティアラ (Golden Tiara)
スチールヘッドギア (Steel Headgear)
プラチナメット (Platinum Helmet)
サタンヘルム (Satan Helm)
武王のはちがね (War King's Headband)
ミスリルフード (Mithril Hood)
スチヌラヘルム (Stenura Helm)
不幸の兜 (Helmet of Misfortune)
はぐれスチルヘルム (Stray Steel Helm)
慈愛の兜 (Helmet of Compassion)
チェインフード (Chain Hood)
鉄兜 (Iron Helmet)
鉄仮面 (Iron Mask)
戦士の兜 (Warrior's Helmet)
猛牛ヘルム (Bull Helm)
玉鋼の兜 (Tamahagane Helmet)
グレートヘルム (Great Helm)
ミスリルヘルム (Mithril Helm)
幻魔の兜 (Phantom Demon Helmet)
女神の兜 (Goddess Helmet)
オルテガの兜 (Ortega's Helmet)
エンデの兜 (Ende's Helmet)
スチルキングヘルム (Steel King Helm)
天空の兜 (Heavenly Helmet)
太陽の冠 (Crown of the Sun)
ロトの兜 (Roto's Helmet)
竜のお守り (Dragon Amulet)
結界の指輪 (Ring of Protection)
護りのルビー (Ruby of Protection)
力のルビー (Ruby of Power)
怒りのタトゥー (Tattoo of Rage)
豪傑の腕輪 (Brave's Bracelet)
はやてのリング (Swift Ring)
疾風のバンダナ (Gale Bandana)
星降る腕輪 (Starfall Bracelet)
インテリめがね (Intelligent Glasses)
博識リング (Knowledge Ring)
知力の腕輪 (Bracelet of Wisdom)
大地の指輪 (Earth Ring)
命の指輪 (Ring of Life)
妖精の腕輪 (Fairy Bracelet)
女神の指輪 (Goddess Ring)
幸運のお守り (Lucky Charm)
ラッキーペンダント (Lucky Pendant)
魔除けの聖印 (Talisman of Protection)
破毒のリング (Antidote Ring)
めざましリング (Awakening Ring)
音溢れの指輪 (Sound Overflow Ring)
理性のリング (Ring of Reason)
エルフのお守り (Elf Charm)
癒しの腕輪 (Healing Bracelet)
祈りの腕輪 (Prayer Bracelet)
静寂の玉 (Orb of Silence)
うさぎのしっぽ (Rabbit's Tail)
銀の指輪 (Silver Ring)
炎のリング (Ring of Fire)
水のリング (Ring of Water)
アバンのしるし (Avan's Emblem)
太陽の指輪 (Ring of the Sun)
月の指輪 (Ring of the Moon)
業師の手袋 (Artisan's Gloves)
達人の手袋 (Master's Gloves)
神業の手袋 (Divine Gloves)
革の靴 (Leather Shoes)
盗賊のブーツ (Thief's Boots)
黄金の靴 (Golden Shoes)
速足のブーツ (Swift Boots)
フェザーシューズ (Feather Shoes)
魔法の靴 (Magic Shoes)
ステルスシューズ (Stealth Shoes)
ヌライムピアス (Nuraim Earrings)
翼のブローチ (Winged Brooch)
世界樹の皮 (World Tree Bark)
ちいさな虫かご (Small Bug Cage)
匂い袋 (Scent Bag)
名人のルアー (Master's Lure)
怪しげなコイン (Suspicious Coin)
謎の鉱物 (Mysterious Mineral)
精霊の腕輪 (Spirit Bracelet)
魔導ユニット (Magic Unit)
竜の角のペンダント (Dragon Horn Pendant)
千里眼の眼鏡 (Clairvoyant Glasses)
幸運の帽子 (Lucky Hat)
修験者の杖 (Ascetic's Staff)
業師の腕輪 (Artisan's Bracelet)
魔物使いの心得 (Beast Tamer's Knowledge)
カイトバックラー (Kite Buckler)
ミスリルシールド (Mithril Shield)
ドルフィンシールド (Dolphin Shield)
聖騎士の盾 (Holy Knight's Shield)
戦士の大盾 (Warrior's Great Shield)
スケイルメイル (Scale Mail)
ホーリーチェイン (Holy Chain)
ルーンスケイル (Rune Scale)
マスターチェイン (Master Chain)
厚手の鎧 (Thick Armor)
ヘヴィアーマー (Heavy Armor)
毛皮のポンチョ (Fur Poncho)
ノーブルマント (Noble Mantle)
ブラックマント (Black Mantle)
マジカルマント (Magical Mantle)
ミラクルマント (Miracle Mantle)
聖者の法衣 (Saint's Robe)
とんがりぼうし (Pointed Hat)
風の帽子 (Wind Hat)
星のサークレット (Star Circlet)
闇夜の帽子 (Dark Night Hat)
アイアンヘッドギア (Iron Headgear)
銀仮面 (Silver Mask)
くろがねのはちがね (Iron Headband)
知力の兜 (Intellect Helm)
ヘヴィヘルム (Heavy Helm)
竜の指輪 (Dragon Ring)
光の指輪 (Light Ring)
闇の指輪 (Dark Ring)
精霊の指輪 (Spirit Ring)
タリホー (Tariho)
```