Update app.py (#15)

- Update app.py (84ba156c4250e9eb6a5e1535e25ec39a19bc42e7)


Co-authored-by: Muhammad Khaqan Nasir <[email protected]>

app.py

@@ -1,52 +1,3 @@
-# import sys
-# from pathlib import Path
-# import os
-# import gdown
-# import streamlit as st
-
-# # Set page config - must be first Streamlit command
-# st.set_page_config(
-#     page_title="TrueCheck - Fake News Detection",
-#     page_icon="📰",
-#     layout="wide"
-# )
-
-# MODEL_PATH = "models/saved/final_model.pt"
-# # You need to replace this with the direct download link of your model file
-# # To get the direct link: Right-click your model file in Google Drive -> Get link -> Make sure it's set to "Anyone with the link can view"
-# # Then replace the file ID in the URL below
-# GOOGLE_DRIVE_FILE_ID = "1xhYKuC5_Yri3mm3Ejt-SpB2WAVUTJvc_"  # Replace with your actual file ID
-# GOOGLE_DRIVE_URL = f"https://drive.google.com/uc?id={GOOGLE_DRIVE_FILE_ID}"
-
-# @st.cache_resource
-# def download_model():
-#     """Download model from Google Drive if not exists."""
-#     if not os.path.exists(MODEL_PATH):
-#         os.makedirs(os.path.dirname(MODEL_PATH), exist_ok=True)
-#         with st.spinner("Downloading model from Google Drive..."):
-#             try:
-#                 gdown.download(GOOGLE_DRIVE_URL, MODEL_PATH, quiet=False)
-#                 st.success("Model downloaded successfully!")
-#             except Exception as e:
-#                 st.error(f"Failed to download model: {str(e)}")
-#                 st.error("Please check your Google Drive link and make sure the file is publicly accessible.")
-#                 return False
-#     return True
-
-# # Add src directory to Python path
-# src_path = Path(__file__).parent / "src"
-# sys.path.append(str(src_path))
-
-# if __name__ == "__main__":
-#     # Download model first
-#     if download_model():
-#         # Import and run the main app
-#         from src.app import main
-#         main()
-#     else:
-#         st.error("Cannot start the application without the model file.") 
-
-
 import sys
 from pathlib import Path
 import os
@@ -55,8 +6,8 @@ import streamlit as st
 
 # Set page config - must be first Streamlit command
 st.set_page_config(
-    page_title="TrueCheck - Fake News Detection",
-    page_icon="📰",
+    page_title="TruthCheck - Advanced Fake News Detector",
+    page_icon="🛡️",
     layout="wide",
     initial_sidebar_state="expanded"
 )
@@ -119,6 +70,29 @@ st.markdown("""
         background-color: #FEE2E2;
         color: #B71C1C;
     }
+    
+    /* Enhanced sidebar styling */
+    .stRadio > div {
+        gap: 0.5rem;
+    }
+    
+    .stRadio > div > label {
+        padding: 0.5rem 1rem;
+        border-radius: 8px;
+        transition: all 0.3s ease;
+        cursor: pointer;
+        font-weight: 500;
+    }
+    
+    .stRadio > div > label:hover {
+        background-color: #E8F0FE;
+        transform: translateX(5px);
+    }
+    
+    .stRadio > div > label[data-checked="true"] {
+        background-color: #4B5EAA;
+        color: white;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -145,38 +119,74 @@ def download_model():
 src_path = Path(__file__).parent / "src"
 sys.path.append(str(src_path))
 
-# Sidebar navigation
-st.sidebar.title("TrueCheck")
+# Enhanced Sidebar navigation with icons
+st.sidebar.markdown("""
+<div style="text-align: center; margin-bottom: 2rem;">
+    <div style="font-size: 2.5rem; margin-bottom: 0.5rem;">🛡️</div>
+    <h1 style="color: #4B5EAA; font-size: 1.5rem; font-weight: 600; margin-bottom: 0.5rem; line-height: 1.2;">
+        TruthCheck
+    </h1>
+    <p style="color: #666; font-size: 0.9rem; margin: 0; font-weight: 300; line-height: 1.3;">
+        Advanced Fake News Detector
+    </p>
+</div>
+""", unsafe_allow_html=True)
+
 st.sidebar.markdown("---")
+
+# Navigation with icons
 page = st.sidebar.radio(
     "Navigate",
-    ["Home", "Team", "About", "Terms of Use", "Privacy Policy"],
-    label_visibility="collapsed"
+    [
+        "🏠 Home",
+        "ℹ️ About", 
+        "📋 Terms of Use",
+        "🔒 Privacy Policy",
+        "👥 Team"
+    ],
+    label_visibility="collapsed",
+    key="navigation"
 )
 
-# SVG icon for sidebar
+# Add some spacing and additional info
+st.sidebar.markdown("---")
 st.sidebar.markdown("""
-<svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="#4B5EAA" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" style="margin-bottom: 1rem;">
-    <path d="M3 9l9-7 9 7v11a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2z"></path>
-    <polyline points="9 22 9 12 15 12 15 22"></polyline>
-</svg>
+<div style="text-align: center; padding: 1.5rem; background: linear-gradient(135deg, #F8F9FA 0%, #E8F0FE 100%); border-radius: 12px; margin-top: 1rem; border: 1px solid #E0E7FF;">
+    <div style="font-size: 2rem; margin-bottom: 0.8rem;">🔍</div>
+    <p style="font-size: 0.85rem; color: #4B5EAA; margin: 0; font-weight: 500; line-height: 1.4;">
+        Protecting you from misinformation with AI-powered detection
+    </p>
+</div>
 """, unsafe_allow_html=True)
 
+# Add footer info
+st.sidebar.markdown("""
+<div style="position: fixed; bottom: 1rem; left: 1rem; right: 1rem; text-align: center; padding: 0.5rem; background-color: rgba(255,255,255,0.9); border-radius: 8px; border-top: 1px solid #E0E0E0;">
+    <p style="font-size: 0.7rem; color: #888; margin: 0;">
+        © 2025 TruthCheck | AI-Powered
+    </p>
+</div>
+""", unsafe_allow_html=True)
+
+# Clean up the page variable to remove icons for the conditional logic
+page_clean = page.split(" ", 1)[1] if " " in page else page
+
 if __name__ == "__main__":
-    if page == "Home":
+    if page_clean == "Home":
         if download_model():
             from src.app import main
             main()
         else:
             st.markdown('<div class="flash-message error-message">Cannot start the application without the model file.</div>', unsafe_allow_html=True)
-    elif page == "About":
+    elif page_clean == "About":
         from src.about import main
         main()
-    elif page == "Terms of Use":
+    elif page_clean == "Terms of Use":
         from src.terms_of_use import main
         main()
-    elif page == "Privacy Policy":
+    elif page_clean == "Privacy Policy":
         from src.privacy_policy import main
-    elif page == "Contact":
-        from src.contact import main
+        main()
+    elif page_clean == "Team":
+        from src.team import main
         main()