Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import reveal_slides as rs
 import time
+from functions import *
 
 # Initialize session state variables if they don't exist
 if 'recap_period' not in st.session_state:
@@ -42,12 +43,9 @@ mock_repos = {
     "private": ["repo_private_1", "repo_private_2", "repo_private_3"]
 }
 
-mock_branches = ["main", "dev", "feature-1", "bugfix-a"]
-mock_contributors = ["Alice", "Bob", "Charlie", "Diana"]
 
 # App title
 st.title("GitHub Contributors Recap")
-st.title(st.secrets["OPEN_AI_TOKEN"])
 
 # Recap period selection
 st.session_state['recap_period'] = st.selectbox("Choose recap period", options=["month", "3 months", "6 months", "year"], index=0)
@@ -66,18 +64,20 @@ if st.button("sync"):
     st.session_state['sync'] = True
 # Load button to fetch repo data (simulated with mock data)
 if st.session_state['sync']:
-    time.sleep(1)
-    st.session_state['selected_branches'] = mock_branches  # Simulate fetching branches
+    contributors_list = get_contributors(repo, updated_token=st.session_state['private_token'])
+    branches_list = get_repo_branches(repo, updated_token=st.session_state['private_token'])
+    
+    st.session_state['selected_branches'] = branches_list
     st.session_state['selected_contributor'] = ''  # Reset selected contributor
     
     # Branch selection
     branch_option = st.radio("Select branches:", options=["all", "specific"])
     
     if branch_option == "specific":
-        st.session_state['selected_branches'] = st.multiselect("Select branches", options=mock_branches, default=mock_branches)
+        st.session_state['selected_branches'] = st.multiselect("Select branches", options=branches_list, default=mock_branches)
     
     # Contributor selection
-    st.session_state['selected_contributor'] = st.selectbox("Select contributor name", options=mock_contributors)
+    st.session_state['selected_contributor'] = st.selectbox("Select contributor name", options=contributors_list)
     
     # Get recap button
     if st.button("Get Recap"):