Feat: adds front and infra

front/app/st_auth.py

@@ -0,0 +1,102 @@
+import streamlit_authenticator as stauth
+import yaml
+from yaml.loader import SafeLoader
+import streamlit as st
+import boto3
+from botocore.exceptions import ClientError
+def get_authenticator():
+    # with open('.streamlit/users.yaml') as file:
+    #     cred_config = yaml.load(file, Loader=SafeLoader)
+
+    file_content = read_text_file_from_s3('chatbot-editais-auth', 'config.yaml')
+    # Parse the YAML content safely
+    cred_config = yaml.safe_load(file_content)
+    
+    # Pre-hashing all plain text passwords once
+    # hash_credentials = stauth.Hasher.hash_passwords(config['credentials'])
+
+    authenticator = stauth.Authenticate(
+        cred_config['credentials'],
+        cred_config['cookie']['name'],
+        cred_config['cookie']['key'],
+        cred_config['cookie']['expiry_days']
+    )
+    return authenticator
+
+def st_authenticate(authenticator: stauth.Authenticate):
+    # authenticator = get_authenticator()
+    try:
+        authenticator.login(
+            location='sidebar',
+            fields=dict(Username="Usuário",
+                        Password="Senha")
+        )
+    except Exception as e:
+        st.error(e)
+        authenticator.cookie_controller.delete_cookie()
+        st.warning('Caso o erro persistir, tente recarregar a página.')
+        st.stop()
+
+    if st.session_state['authentication_status']:
+        with st.sidebar:
+            st.write(f'Usuário: {st.session_state["name"]}')
+            authenticator.logout()
+    elif st.session_state['authentication_status'] is False:
+        with st.sidebar:
+            st.error('Usuário/senha incorreto')
+        st.stop()
+    elif st.session_state['authentication_status'] is None:
+        st.warning('Por favor, informe o seu usuário e senha no painel lateral')
+        st.stop()
+def read_text_file_from_s3(bucket, key):
+    """
+    Read a YAML file from an S3 bucket using provided AWS credentials.
+    
+    Args:
+        bucket (str): Name of the S3 bucket
+        key (str): Path to the YAML file in the bucket
+        aws_access_key_id (str): AWS Access Key ID
+        aws_secret_access_key (str): AWS Secret Access Key
+    
+    Returns:
+        dict: Parsed YAML content
+    """
+    try:
+        # Create an S3 client
+        s3_client = boto3.client('s3')
+        
+        # Download the file from S3
+        response = s3_client.get_object(Bucket=bucket, Key=key)
+        
+        # Read the file content
+        file_content = response['Body'].read().decode('utf-8')
+        
+        return file_content
+    
+    except Exception as e:
+        st.error(f"Error reading file from S3: {e}")
+        return None
+    
+def get_secret():
+
+    secret_name = "dev/chatboteditais/apigateway/apikey"
+    region_name = "us-east-1"
+
+    # Create a Secrets Manager client
+    session = boto3.session.Session()
+    client = session.client(
+        service_name='secretsmanager',
+        region_name=region_name
+    )
+
+    try:
+        get_secret_value_response = client.get_secret_value(
+            SecretId=secret_name
+        )
+    except ClientError as e:
+        # For a list of exceptions thrown, see
+        # https://docs.aws.amazon.com/secretsmanager/latest/apireference/API_GetSecretValue.html
+        raise e
+
+    secret = get_secret_value_response['SecretString']
+    return secret