Update app.py

app.py

@@ -1,16 +1,23 @@
 import streamlit as st
 import pandas as pd
 import os
-from utils.search import get_collaboration_data
+import sys
 
-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
 
-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
+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)
+os.makedirs(data_path, exist_ok=True)
+os.makedirs(results_path, exist_ok=True)
 
 # ✅ Load competitor list using absolute path
-base_path = os.path.dirname(os.path.abspath(__file__))
-csv_path = os.path.join(base_path, "data", "competitors.csv")
+csv_path = os.path.join(data_path, "competitors.csv")
 
 try:
     competitors_df = pd.read_csv(csv_path)
@@ -27,8 +34,8 @@ 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:
-            results_path = os.path.join(base_path, "results", "virgin_collab_results.csv")
-            result_df.to_csv(results_path, index=False)
+            result_file_path = os.path.join(results_path, "virgin_collab_results.csv")
+            result_df.to_csv(result_file_path, index=False)
             st.success("✅ Search Completed!")
             st.dataframe(result_df)
         else:
@@ -36,9 +43,9 @@ if st.button("🔍 Run Collaboration Search"):
 
 # ✅ Show saved results
 if st.button("📂 Show Saved Results"):
-    results_path = os.path.join(base_path, "results", "virgin_collab_results.csv")
+    result_file_path = os.path.join(results_path, "virgin_collab_results.csv")
     try:
-        saved_df = pd.read_csv(results_path)
+        saved_df = pd.read_csv(result_file_path)
         if saved_df.empty:
             st.warning("⚠️ The saved file is empty. Try running the search again.")
         else: