llarga/app.py at main · dhopp1/llarga · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import streamlit as st
import time
import torch

torch.classes.__path__ = []  # necessary for Docker

from helper.sidebar import (
    sidebar_batch_query,
    sidebar_chats,
    sidebar_cite_sources,
    sidebar_delete_corpus,
    sidebar_export_chat,
    gen_export_df,
    sidebar_llm_api_key,
    sidebar_llm_dropdown,
    sidebar_temperature_dropdown,
    sidebar_stop_llamacpp,
    sidebar_system_prompt,
    sidebar_upload_file,
    sidebar_web_search,
    sidebar_which_corpus,
)
from helper.user_management import check_password, setup_local_files
from helper.ui import (
    import_chat,
    import_styles,
    initial_placeholder,
    metadata_tab,
    ui_title_icon,
)


# load user list and llm list
setup_local_files()
ui_title_icon()

# login screen
if not check_password():
    st.stop()

### initial setup
# styles sheets
import_styles()

# placeholder on initial load
initial_placeholder()

### sidebar
st.sidebar.markdown(f'# {st.session_state["user_name"]}')
# previous chats
st.sidebar.markdown("### Chats")
sidebar_chats()

# corpus info
st.sidebar.markdown("### Corpus")
with st.sidebar:
    sidebar_which_corpus()
    gen_export_df()
    with st.expander("Corpus metadata"):
        metadata_tab()

with st.sidebar.expander("Upload your own documents"):
    st.markdown("#### Upload a new corpus")
    sidebar_upload_file()
    st.markdown("#### Delete a corpus")
    sidebar_delete_corpus()

# llm info
st.sidebar.markdown("### LLM")
with st.sidebar:
    sidebar_web_search()
    sidebar_cite_sources()

with st.sidebar.expander("LLM parameters"):
    sidebar_llm_dropdown()
    sidebar_llm_api_key()
    sidebar_temperature_dropdown()
    sidebar_system_prompt()
    sidebar_stop_llamacpp()

# warning if system prompt is different different
with st.sidebar:
    if (
        st.session_state["default_system_prompt"] != st.session_state["system_prompt"]
    ) and st.session_state["selected_corpus"] != "No corpus":
        st.warning(
            "The default system prompt for this corpus differs from what you have input in `System prompt` under the `LLM parameters` dropdown. Consider changing it. The default system prompt for this corpus is:"
        )
        st.markdown(f"""```\n{st.session_state["default_system_prompt"]}\n```""")

# batch query
with st.sidebar.expander("Batch query"):
    sidebar_batch_query()

### chat logic
import_chat()

# export chat button
with st.sidebar:
    sidebar_export_chat()

    # logout
    def logout():
        st.sidebar.info(
            "The next time you refresh the page you will need to sign in again."
        )
        time.sleep(3)
        st.session_state["cookie_manager"].delete(cookie="logged_in")
        st.session_state["cookie_manager"].delete(cookie="username")

    st.button("Logout", on_click=logout)