Émile
commited on
Commit
·
17b81f2
1
Parent(s):
7e5c318
Switching logs to terminal, personnal infos hidden
Browse files
app.py
CHANGED
@@ -109,7 +109,11 @@ llm = setup_generator("gpt-4o-mini", max_tokens=30)
|
|
109 |
llm2 = setup_generator("gpt-4o-mini")
|
110 |
|
111 |
|
112 |
-
embedder = SentenceTransformersTextEmbedder(
|
|
|
|
|
|
|
|
|
113 |
retriever = ChromaEmbeddingRetriever(document_store)
|
114 |
|
115 |
router = ConditionalRouter(routes=routes)
|
@@ -175,22 +179,10 @@ def hide_sensitive_info(message):
|
|
175 |
return text
|
176 |
return "[Error when hiding user info, no log generated]"
|
177 |
|
178 |
-
publicKey = rsa.key.PublicKey(12771964615703412689771875940203228650714115641982293999355383771918953923930646807301050027636271229722536699443183546682257910554154445751549366908636779, 65537)
|
179 |
-
|
180 |
-
def encripted_log(message: str):
|
181 |
-
fernet_key = Fernet.generate_key()
|
182 |
-
fernet = Fernet(fernet_key)
|
183 |
-
encrypted_key = rsa.encrypt(fernet_key, publicKey)
|
184 |
-
return (encrypted_key.hex() + fernet.encrypt(message.encode()).hex())
|
185 |
-
|
186 |
-
|
187 |
def log_QA(question, answer):
|
188 |
message = f"User: {question}\nAssistant: {answer}"
|
189 |
message_no_info = hide_sensitive_info(message)
|
190 |
-
|
191 |
-
with open("log.txt", "a") as f:
|
192 |
-
f.write(encripted_QA + "\n")
|
193 |
-
|
194 |
|
195 |
##########################
|
196 |
####### Gradio app #######
|
|
|
109 |
llm2 = setup_generator("gpt-4o-mini")
|
110 |
|
111 |
|
112 |
+
embedder = SentenceTransformersTextEmbedder(
|
113 |
+
model=embedding_model,
|
114 |
+
trust_remote_code=True,
|
115 |
+
progress_bar=False
|
116 |
+
)
|
117 |
retriever = ChromaEmbeddingRetriever(document_store)
|
118 |
|
119 |
router = ConditionalRouter(routes=routes)
|
|
|
179 |
return text
|
180 |
return "[Error when hiding user info, no log generated]"
|
181 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
182 |
def log_QA(question, answer):
|
183 |
message = f"User: {question}\nAssistant: {answer}"
|
184 |
message_no_info = hide_sensitive_info(message)
|
185 |
+
print(message_no_info, end="\n\n")
|
|
|
|
|
|
|
186 |
|
187 |
##########################
|
188 |
####### Gradio app #######
|
log.txt
DELETED
File without changes
|