Hugging Face's logo

Spaces:
RustX
/
CSV-ChatBot
Runtime error

App Files Community
CSV-ChatBot
File size: 2,432 Bytes 
19ef06b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
import os
import pandas as pd
import streamlit as st

from modules.chatbot import Chatbot
from modules.embedder import Embedder


class Utilities:
    @staticmethod
    def load_api_key():
        """
        Loads the OpenAI API key from the .env file or from the user's input
        and returns it
        """
        if os.path.exists(".env") and os.environ.get("OPENAI_API_KEY") is not None:
            user_api_key = os.environ["OPENAI_API_KEY"]
            st.sidebar.success("API key loaded from .env / .envμ—μ„œ λ‘œλ“œλœ API ν‚€", icon="πŸš€")
        else:
            user_api_key = st.sidebar.text_input(
                label="#### Your OpenAI API key / OpenAI API ν‚€ πŸ‘‡", placeholder="Paste your openAI API key, sk-", type="password"
            )
            if user_api_key:
                st.sidebar.success("API key loaded / API ν‚€κ°€ λ‘œλ“œλ˜μ—ˆμŠ΅λ‹ˆλ‹€", icon="πŸš€")
        return user_api_key

    @staticmethod
    def handle_upload():
        """
        Handles the file upload and displays the uploaded file
        """
        uploaded_file = st.sidebar.file_uploader("upload", type="csv", label_visibility="collapsed")
        if uploaded_file is not None:

            def show_user_file(uploaded_file):
                file_container = st.expander("Your CSV file : / CSV 파일:")
                shows = pd.read_csv(uploaded_file)
                uploaded_file.seek(0)
                file_container.write(shows)

            show_user_file(uploaded_file)
        else:
            st.sidebar.info(
                "πŸ‘† Upload your CSV file to get started, / μ‹œμž‘ν•˜λ €λ©΄ CSV νŒŒμΌμ„ μ—…λ‘œλ“œν•˜μ„Έμš” "
                "sample for try : / μ‹œλ„ν•΄ λ³Ό μƒ˜ν”Œ: [example.csv](https://drive.google.com/file/d/1g7x0Ydg5kr51Ha2XIYBSQBVUw1yYlgmc/view?usp=share_link)"
            )
            st.session_state["reset_chat"] = True
        return uploaded_file

    @staticmethod
    def setup_chatbot(uploaded_file, model, temperature):
        """
        Sets up the chatbot with the uploaded file, model, and temperature
        """
        embeds = Embedder()
        with st.spinner("Processing... / 처리 쀑..."):
            uploaded_file.seek(0)
            file = uploaded_file.read()
            vectors = embeds.getDocEmbeds(file, uploaded_file.name)
            chatbot = Chatbot(model, temperature, vectors)
        st.session_state["ready"] = True
        return chatbot