Émile commited on
Commit
17b81f2
·
1 Parent(s): 7e5c318

Switching logs to terminal, personnal infos hidden

Browse files
Files changed (2) hide show
  1. app.py +6 -14
  2. log.txt +0 -0
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(model=embedding_model, trust_remote_code=True)
 
 
 
 
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
- encripted_QA = encripted_log(message_no_info)
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