Update app.py

app.py

@@ -1,7 +1,7 @@
 
 import streamlit as st
-from utils import set_page_config, display_sidebar
-import os
+
+from utils import set_page_config
 
 # Set the Streamlit page configuration
 set_page_config()
@@ -12,28 +12,30 @@ st.title("CodeGen Hub")
 # App description with markdown formatting
 st.markdown("""
     Welcome to CodeGen Hub - A platform for training and using code generation models with Hugging Face integration.
-    
+
     ### Core Features:
     - 📂 Upload and preprocess Python code datasets for model training
     - 🎛️ Configure and train models with customizable parameters
     - 🤖 Generate code predictions using trained models through an interactive interface
     - 📊 Monitor training progress with visualizations and detailed logs
     - 🔗 Seamless integration with Hugging Face Hub for model management
-    
+
     Navigate through the different sections using the sidebar menu.
 """)
 
+
 # Sidebar navigation using session state
 def navigate(page):
     st.session_state["current_page"] = page
 
+
 # Initialize session state variables using a loop
 session_defaults = {
-    "datasets": {},        # Stores uploaded datasets
+    "datasets": {},  # Stores uploaded datasets
     "trained_models": {},  # Stores trained model details
-    "training_logs": [],   # Stores training logs
+    "training_logs": [],  # Stores training logs
     "training_progress": {},  # Tracks active training jobs
-    "current_page": "home"  # Default landing page
+    "current_page": "home",  # Default landing page
 }
 
 for key, value in session_defaults.items():
@@ -81,12 +83,15 @@ st.subheader("Platform Statistics")
 col1, col2, col3 = st.columns(3)
 
 with col1:
-    st.metric("📂 Datasets Available", len(st.session_state.get("datasets", {})))
+    st.metric("📂 Datasets Available", len(st.session_state.get("datasets",
+                                                               {})))
 
 with col2:
-    st.metric("📦 Trained Models", len(st.session_state.get("trained_models", {})))
+    st.metric("📦 Trained Models",
+              len(st.session_state.get("trained_models", {})))
 
 with col3:
-    active_jobs = sum(1 for progress in st.session_state["training_progress"].values() 
-                      if progress.get("status") == "running")
+    active_jobs = sum(
+        1 for progress in st.session_state["training_progress"].values()
+        if progress.get("status") == "running")
     st.metric("🚀 Active Training Jobs", active_jobs)