fix: not able to find prompt directory for English

1 year ago · 7e59af08e3
parent 9b20932325
commit 7e59af08e3
5 changed files with 77 additions and 68 deletions
--- a/.locals/en_US/LC_MESSAGES/base.po
+++ b/.locals/en_US/LC_MESSAGES/base.po
@ -7,7 +7,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: \n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-02-19 03:00+1100\n"
+"POT-Creation-Date: 2023-02-19 03:16+1100\n"
 "PO-Revision-Date: 2023-02-18 23:31+1100\n"
 "Last-Translator: \n"
 "Language-Team: \n"
@ -80,7 +80,7 @@ msgstr "Create"
 msgid "my-info"
 msgstr "my-info"

-#: pages/1_Configs.py:235 Seanium_brain.py:118 Seanium_brain.py:183
+#: pages/1_Configs.py:235 Seanium_brain.py:117 Seanium_brain.py:182
 #, fuzzy
 msgid "question"
 msgstr "Question Model"
@ -166,113 +166,113 @@ msgstr ""
 msgid "Thinking on"
 msgstr "Thinking on Answer"

-#: Seanium_brain.py:110
+#: Seanium_brain.py:109
 msgid "Operations"
 msgstr "Operations"

-#: Seanium_brain.py:116
+#: Seanium_brain.py:115
 msgid "Question Model"
 msgstr "Question Model"

-#: Seanium_brain.py:125
+#: Seanium_brain.py:124
 msgid "Model"
 msgstr ""

-#: Seanium_brain.py:128
+#: Seanium_brain.py:127
 msgid "Temperature"
 msgstr "Temperature"

-#: Seanium_brain.py:129
+#: Seanium_brain.py:128
 msgid "Max Tokens"
 msgstr "Max Tokens"

-#: Seanium_brain.py:131
+#: Seanium_brain.py:130
 msgid "Advanced Options"
 msgstr "Advanced Options"

-#: Seanium_brain.py:132
+#: Seanium_brain.py:131
 msgid "Top_P"
 msgstr ""

-#: Seanium_brain.py:133
+#: Seanium_brain.py:132
 msgid "Frequency penalty"
 msgstr "Frequency penalty"

-#: Seanium_brain.py:135
+#: Seanium_brain.py:134
 msgid "Presence penalty"
 msgstr "Presence penalty"

-#: Seanium_brain.py:138
+#: Seanium_brain.py:137
 msgid "Chunk size"
 msgstr "Chunk size"

-#: Seanium_brain.py:139
+#: Seanium_brain.py:138
 msgid "Answer count"
 msgstr "Answer count"

-#: Seanium_brain.py:149
+#: Seanium_brain.py:148
 msgid "Clear Log"
 msgstr "Clear Log"

-#: Seanium_brain.py:150
+#: Seanium_brain.py:149
 msgid "Log Cleared"
 msgstr "Log Cleared"

-#: Seanium_brain.py:155
+#: Seanium_brain.py:154
 msgid "version"
 msgstr ""

-#: Seanium_brain.py:156
+#: Seanium_brain.py:155
 msgid "author"
 msgstr ""

-#: Seanium_brain.py:157
+#: Seanium_brain.py:156
 msgid "Report bugs"
 msgstr ""

-#: Seanium_brain.py:158
+#: Seanium_brain.py:157
 msgid "Github Repo"
 msgstr ""

-#: Seanium_brain.py:161
+#: Seanium_brain.py:160
 #, fuzzy
 msgid "🧠GPT-Brain"
 msgstr "🧠Seanium Brain"

-#: Seanium_brain.py:162
+#: Seanium_brain.py:161
 msgid ""
 "This is my personal AI powered brain feeding my own Obsidian notes. Ask "
 "anything."
 msgstr ""

-#: Seanium_brain.py:164
+#: Seanium_brain.py:163
 msgid "This is a beta version. Please [🪲report bugs]("
 msgstr ""

-#: Seanium_brain.py:165
+#: Seanium_brain.py:164
 msgid ") if you find any."
 msgstr ""

-#: Seanium_brain.py:173
+#: Seanium_brain.py:172
 msgid "Building Brain..."
 msgstr "Building Brain..."

-#: Seanium_brain.py:176
+#: Seanium_brain.py:175
 msgid "Brain rebuild!"
 msgstr "Brain rebuild!"

-#: Seanium_brain.py:180
+#: Seanium_brain.py:179
 msgid "Thinking on Answer"
 msgstr "Thinking on Answer"

-#: Seanium_brain.py:185
+#: Seanium_brain.py:184
 msgid "💬Answer"
 msgstr "💬Answer"

-#: Seanium_brain.py:217
+#: Seanium_brain.py:216
 msgid "Ask Brain: "
 msgstr "Ask Brain: "

-#: Seanium_brain.py:220
+#: Seanium_brain.py:219
 msgid "📩Send"
 msgstr "📩Send"
--- a/.locals/zh_CN/LC_MESSAGES/base.po
+++ b/.locals/zh_CN/LC_MESSAGES/base.po
@ -7,8 +7,8 @@ msgid ""
 msgstr ""
 "Project-Id-Version: \n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2023-02-19 03:00+1100\n"
-"PO-Revision-Date: 2023-02-19 03:01+1100\n"
+"POT-Creation-Date: 2023-02-19 03:16+1100\n"
+"PO-Revision-Date: 2023-02-19 03:16+1100\n"
 "Last-Translator: \n"
 "Language-Team: \n"
 "Language: zh_CN\n"
@ -79,7 +79,7 @@ msgstr "创建"
 msgid "my-info"
 msgstr "我的背景"

-#: pages/1_Configs.py:235 Seanium_brain.py:118 Seanium_brain.py:183
+#: pages/1_Configs.py:235 Seanium_brain.py:117 Seanium_brain.py:182
 msgid "question"
 msgstr "问题"

@ -163,113 +163,113 @@ msgstr "正在处理"
 msgid "Thinking on"
 msgstr "正在思考"

-#: Seanium_brain.py:110
+#: Seanium_brain.py:109
 msgid "Operations"
 msgstr "操作"

-#: Seanium_brain.py:116
+#: Seanium_brain.py:115
 msgid "Question Model"
 msgstr "问题模型"

-#: Seanium_brain.py:125
+#: Seanium_brain.py:124
 msgid "Model"
 msgstr "模型"

-#: Seanium_brain.py:128
+#: Seanium_brain.py:127
 msgid "Temperature"
 msgstr "温度（Temperature）"

-#: Seanium_brain.py:129
+#: Seanium_brain.py:128
 msgid "Max Tokens"
 msgstr "最大令牌数（Max Token）"

-#: Seanium_brain.py:131
+#: Seanium_brain.py:130
 msgid "Advanced Options"
 msgstr "高级设置"

-#: Seanium_brain.py:132
+#: Seanium_brain.py:131
 msgid "Top_P"
 msgstr ""

-#: Seanium_brain.py:133
+#: Seanium_brain.py:132
 msgid "Frequency penalty"
 msgstr "频率惩罚（Frequency penalty）"

-#: Seanium_brain.py:135
+#: Seanium_brain.py:134
 msgid "Presence penalty"
 msgstr "存在惩罚（Presence penalty）"

-#: Seanium_brain.py:138
+#: Seanium_brain.py:137
 msgid "Chunk size"
 msgstr "区块大小（Chunk size）"

-#: Seanium_brain.py:139
+#: Seanium_brain.py:138
 msgid "Answer count"
 msgstr "回答数量（Answer count）"

-#: Seanium_brain.py:149
+#: Seanium_brain.py:148
 msgid "Clear Log"
 msgstr "清除日志"

-#: Seanium_brain.py:150
+#: Seanium_brain.py:149
 msgid "Log Cleared"
 msgstr "日志以清除"

-#: Seanium_brain.py:155
+#: Seanium_brain.py:154
 msgid "version"
 msgstr "版本"

-#: Seanium_brain.py:156
+#: Seanium_brain.py:155
 msgid "author"
 msgstr "作者"

-#: Seanium_brain.py:157
+#: Seanium_brain.py:156
 msgid "Report bugs"
 msgstr "报告bug"

-#: Seanium_brain.py:158
+#: Seanium_brain.py:157
 msgid "Github Repo"
 msgstr "Github源代码"

-#: Seanium_brain.py:161
+#: Seanium_brain.py:160
 msgid "🧠GPT-Brain"
 msgstr "🧠GPT-大脑"

-#: Seanium_brain.py:162
+#: Seanium_brain.py:161
 msgid ""
 "This is my personal AI powered brain feeding my own Obsidian notes. Ask "
 "anything."
 msgstr "这是我的个人AI知识管理库，请随意问。"

-#: Seanium_brain.py:164
+#: Seanium_brain.py:163
 msgid "This is a beta version. Please [🪲report bugs]("
 msgstr "该版本为BETA测试版。如果遇到BUG，请[🪲在此处报告BUG]("

-#: Seanium_brain.py:165
+#: Seanium_brain.py:164
 msgid ") if you find any."
 msgstr ")。"

-#: Seanium_brain.py:173
+#: Seanium_brain.py:172
 msgid "Building Brain..."
 msgstr "重建大脑内容"

-#: Seanium_brain.py:176
+#: Seanium_brain.py:175
 msgid "Brain rebuild!"
 msgstr "大脑已重建！"

-#: Seanium_brain.py:180
+#: Seanium_brain.py:179
 msgid "Thinking on Answer"
 msgstr "思考答案中"

-#: Seanium_brain.py:185
+#: Seanium_brain.py:184
 msgid "💬Answer"
 msgstr "💬回答"

-#: Seanium_brain.py:217
+#: Seanium_brain.py:216
 msgid "Ask Brain: "
 msgstr "提问大脑："

-#: Seanium_brain.py:220
+#: Seanium_brain.py:219
 msgid "📩Send"
 msgstr "📩发送"

--- a/Seanium_Brain.py
+++ b/Seanium_Brain.py
@ -11,8 +11,7 @@ import os
 # activate session
 if 'SESSION_TIME' not in st.session_state:
    st.session_state['SESSION_TIME'] = time.strftime("%Y%m%d-%H%H%S")
-if 'SESSION_LANGUAGE' not in st.session_state:
-    st.session_state['SESSION_LANGUAGE'] = util.read_json_at('.user/language.json', 'SESSION_LANGUAGE', 'en_US')
+

 st.set_page_config(
    page_title='GPT Brain'
--- a/brain.py
+++ b/brain.py
@ -2,12 +2,18 @@ import openai
 import textwrap

 from modules import utilities as util
+from modules import language
+import streamlit as st
 from modules import gpt_util as gpt

-
 openai.api_key = util.read_file(r'.user\API-KEYS.txt').strip()

-prompt_dir = '.user/prompt'
+if 'SESSION_LANGUAGE' not in st.session_state:
+    st.session_state['SESSION_LANGUAGE'] = util.read_json_at('.user/language.json', 'SESSION_LANGUAGE', 'en_US')
+
+SESSION_LANG = st.session_state['SESSION_LANGUAGE']
+prompt_dir = f'.user/prompt/{SESSION_LANG}'
+_ = language.set_language()


 def build(chunk_size=4000):
@ -33,9 +39,9 @@ def run_answer(query, model, temp, max_tokens, top_p, freq_penl, pres_penl, chun
    results = gpt.search_chunks(query, brain_data, chunk_count)
    answers = []
    for result in results:
-        my_info = util.read_file(f'{prompt_dir}/my-info.txt')
+        my_info = util.read_file(f'{prompt_dir}/' + _('my-info') + '.txt')

-        prompt = util.read_file(f'{prompt_dir}/question.txt')
+        prompt = util.read_file(f'{prompt_dir}/' + _('question') + '.txt')
        prompt = prompt.replace('<<INFO>>', result['content'])
        prompt = prompt.replace('<<QS>>', query)
        prompt = prompt.replace('<<MY-INFO>>', my_info)
--- a/modules/gpt_util.py
+++ b/modules/gpt_util.py
@ -1,17 +1,20 @@
 import openai
 import numpy as np

-# this function compare similarity between two vectors. 
+
+# this function compare similarity between two vectors.
 # The higher value the dot product have, the more alike between these vectors
 def similarity(v1, v2):
    return np.dot(v1, v2)

+
 # return a list of vectors
 def embedding(content, engine='text-embedding-ada-002'):
    response = openai.Embedding.create(input=content, engine=engine)
    vector = response['data'][0]['embedding']
    return vector

+
 def search_chunks(text, data, count=1):
    vector = embedding(text)
    points = []
@ -28,9 +31,10 @@ def search_chunks(text, data, count=1):

    return ordered[0:count]

+
 def gpt3(prompt, model, temp, max_tokens, top_p, freq_penl, pres_penl):
    response = openai.Completion.create(
-        model= model,
+        model=model,
        prompt=prompt,
        temperature=temp,
        max_tokens=max_tokens,
@ -39,4 +43,4 @@ def gpt3(prompt, model, temp, max_tokens, top_p, freq_penl, pres_penl):
        presence_penalty=pres_penl
    )
    text = response['choices'][0]['text'].strip()
-    return text
+    return text