Spaces:
Sleeping
Sleeping
File size: 1,555 Bytes
9f396ec |
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 |
import streamlit as st
from configfile import Config # Import the Config class
class LoadStreamlitUI:
def __init__(self):
self.config = Config() # Create a Config instance
self.user_controls = {}
def load_streamlit_ui(self):
st.set_page_config(page_title= "🤖 " + self.config.get_page_title(), layout="wide")
st.header("🤖 " + self.config.get_page_title())
with st.sidebar:
# Use case selection
usecase_options = self.config.get_usecase_options()
self.user_controls["selected_usecase"] = st.selectbox("Select Usecases", usecase_options)
# Get options from config
llm_options = self.config.get_llm_options()
self.user_controls["selected_llm"] = st.selectbox("", llm_options)
# model selection
if self.user_controls["selected_usecase"] == "Text Generation":
model_options = self.config.get_text_hf_model_options()
self.user_controls["selected_hf_model"] = st.selectbox("Select Model", model_options)
elif self.user_controls["selected_usecase"] == "Image Generation":
model_options = self.config.get_img_hf_model_options()
self.user_controls["selected_hf_model"] = st.selectbox("Select Model", model_options)
# API key input
st.session_state['api_key'] = st.text_input("API Key",type="password")
return self.user_controls
|