import streamlit as st
import streamlit.components.v1 as components
import os
import time
import pandas as pd

from run_prothgt_app import *

def convert_df(df):
   return df.to_csv(index=False).encode('utf-8')

with st.sidebar:
    st.title("ProtHGT: Heterogeneous Graph Transformers for Automated Protein Function Prediction Using Knowledge Graphs and Language Models")
    st.write("[![publication](https://img.shields.io/badge/DOI-10.1002/pro.4988-b31b1b.svg)]() [![github-repository](https://img.shields.io/badge/GitHub-black?logo=github)](https://github.com/HUBioDataLab/ProtHGT)")

    # Add protein selection
    # You'll need to replace this with your actual data loading
    available_proteins = get_available_proteins()  # Function to get list of proteins from your data
    selected_protein = st.selectbox(
        "Select or search for a protein (UniProt ID)",
        options=available_proteins,
        placeholder="Start typing to search...",
    )
    
    if selected_protein:
        st.write(f"Selected protein: {selected_protein}")