Update app.py

app.py

@@ -15,8 +15,44 @@ from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 from langchain_community.llms import HuggingFaceEndpoint
 import torch
+import os
+import sys
+import getpass
+
+def masked_input(prompt=''):
+    import sys
+    import tty
+    import termios
+    
+    print(prompt, end='', flush=True)
+    fd = sys.stdin.fileno()
+    old_settings = termios.tcgetattr(fd)
+    try:
+        tty.setraw(fd)
+        password = ''
+        while True:
+            ch = sys.stdin.read(1)
+            if ch == '\n' or ch == '\r':
+                break
+            elif ch == '\x7f':  # Handle backspace
+                if len(password) > 0:
+                    password = password[:-1]
+                    sys.stdout.write('\b \b')
+            else:
+                password += ch
+                sys.stdout.write('*')
+            sys.stdout.flush()
+    finally:
+        termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+    print()  # For newline
+    return password
+
+# Solicita ao usuário a chave API de forma segura
+api_key = masked_input(prompt="Digite sua chave API: ")
+os.environ["OPENAI_API_KEY"] = api_key
+# Solicita ao usuário a chave API de forma segura
+
 
-api_token = os.getenv("HF_TOKEN")
 
 list_llm = ["meta-llama/Meta-Llama-3-8B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"]  
 list_llm_simple = [os.path.basename(llm) for llm in list_llm]