import streamlit as st
from src.functions_db import User, Activity, connect_to_db
import pandas as pd

st.title("HQ app E & N")

st.write("Welcome to the app!")

st.write('Some pipelines for the RAG-project are:')
st.checkbox('1. Create and test LLM.')
st.checkbox('2. DB creating and connection.')       
st.checkbox('3. Simple screapper realisation(not so important).')
st.checkbox('4. PDF-reader and analyser (with 3 diff methods).')
st.checkbox('5. API testing and implementing.')
st.checkbox('6. Final realisation of the RAG')


session, _, _, _, _  = connect_to_db(address='sqlite:///src/databases/main.db') # connect to the database

activities = session.query(Activity).all()
users = session.query(User).all()

st.write("Users:")
for user in users:
    st.write(user)

activities_df = pd.DataFrame([{
    'id': activity.id,
    'date': activity.date,
    'time': activity.time,
    'ip': 'hided', #activity.ip,
    'user_id': activity.user_id,
    'type_of_activity': activity.type_of_activity
} for activity in activities])

st.write("Activities:")
st.data_editor(activities_df)