Update app.py

app.py

@@ -1,51 +1,53 @@
+import os
 import streamlit as st
 import pandas as pd
-import os
-import sys
+from utils.search import get_collaboration_data  # Ensure the search.py is correctly placed in utils folder
+
+# Set the page configuration
+st.set_page_config(page_title="Virgin Media Collaboration Finder", layout="centered")
 
-# ✅ Fix Python path for Hugging Face + VS Code (handling imports correctly)
-sys.path.append(os.path.join(os.path.dirname(__file__), 'utils'))  # Ensure the utils directory is found
-from search import get_collaboration_data  # Adjusted import for search.py from utils
+# Page title
+st.title("🕵️ Virgin Media - Competitor Collaboration Finder")
+st.markdown("Use this tool to find which competitors have worked with **Virgin Media**.")
 
-# ✅ Hugging Face File Paths
+# Ensure paths are handled properly, especially on Hugging Face or local environments
 base_path = os.path.dirname(os.path.abspath(__file__))
 data_path = os.path.join(base_path, "data")
 results_path = os.path.join(base_path, "results")
 
-# ✅ Ensure the required directories exist (on Hugging Face Spaces, directories like "data" or "results" may not exist)
+# Ensure the required directories exist (on Hugging Face Spaces, directories like "data" or "results" may not exist)
 os.makedirs(data_path, exist_ok=True)
 os.makedirs(results_path, exist_ok=True)
 
-# ✅ Load competitor list using absolute path
+# Load competitor list using absolute path
 csv_path = os.path.join(data_path, "competitors.csv")
 
 try:
     competitors_df = pd.read_csv(csv_path)
     competitors = competitors_df['Company'].dropna().tolist()
 except FileNotFoundError:
-    st.error("❌ File not found: data/competitors.csv")
+    st.error(f"❌ File not found: {csv_path}")  # Show the correct path
     st.stop()
 except pd.errors.EmptyDataError:
-    st.error("❌ The competitors.csv file is empty.")
+    st.error(f"❌ The competitors.csv file is empty: {csv_path}")
     st.stop()
 
-# ✅ Run search
+# Run search when button is clicked
 if st.button("🔍 Run Collaboration Search"):
     with st.spinner("Searching... This may take a few moments..."):
         result_df = get_collaboration_data(competitors)
         if not result_df.empty:
-            result_file_path = os.path.join(results_path, "virgin_collab_results.csv")
-            result_df.to_csv(result_file_path, index=False)
+            result_csv_path = os.path.join(results_path, "virgin_collab_results.csv")
+            result_df.to_csv(result_csv_path, index=False)
             st.success("✅ Search Completed!")
             st.dataframe(result_df)
         else:
             st.warning("⚠️ No collaboration data found. Nothing was saved.")
 
-# ✅ Show saved results
+# Show saved results when button is clicked
 if st.button("📂 Show Saved Results"):
-    result_file_path = os.path.join(results_path, "virgin_collab_results.csv")
     try:
-        saved_df = pd.read_csv(result_file_path)
+        saved_df = pd.read_csv(os.path.join(results_path, "virgin_collab_results.csv"))
         if saved_df.empty:
             st.warning("⚠️ The saved file is empty. Try running the search again.")
         else: