app.py

import streamlit as st
import pickle
import gdown

# Modules
from FinPredictorX.utils.about import explainAboutApp
from FinPredictorX.price import price

# Download gdown file
@st.cache_resource
def load_models():
    gdown.download(f"https://drive.google.com/uc?id={st.secrets['INDIAN_COMPANIES']}", 'indian_companies.pkl', quiet=False)
    gdown.download(f"https://drive.google.com/uc?id={st.secrets['CRYPTO_CURRENCIES']}", 'crypto_currencies.pkl', quiet=False)

stock_files = {
    "Indian Companies": "indian_companies.pkl",
    "Cryptocurrency": "crypto_currencies.pkl"
}

features = {
    "Price Prediction": price
}

def finPredictorX(feature: str):
    choice = st.sidebar.radio("Select your choice", ["About"] + list(stock_files.keys()))

    if choice == "About":
        explainAboutApp(feature=feature)
        return

    # Load the coins/stocks dataset
    with open(stock_files[choice], "rb") as file:
        stocks = pickle.load(file)

    selected_db = st.sidebar.selectbox(f"Select the {choice}", stocks)

    # Display the selected dataset
    chosen_feature_func = features[feature]
    chosen_feature_func(stocks[selected_db])

if __name__ == '__main__':
    st.title("🌟 Welcome to FinPredictorX App")
    st.sidebar.title("FinPredictorX App")
    load_models()
    feature = st.sidebar.radio("Select the Prediction Type", features.keys())
    if feature:
        finPredictorX(feature=feature)
    else:
        st.sidebar.warning("Select a dataset to start the application!", icon="⚠️")