Update search.py

search.py

@@ -3,16 +3,17 @@ import pandas as pd
 import time
 import os
 
-# ✅ Safely get the API key from the environment or fallback to empty string
-API_KEY = os.getenv("3696a83abc995ae3bb2f392fd9b899d721e7b35b8b29f80cb0bd7b62c55fe132", "")
+# ✅ Retrieve the API key from Hugging Face secrets
+API_KEY = os.getenv("SERPAPI_API_KEY", "")  # This will get the API key from Hugging Face secrets
 
-# ✅ Ensure API_KEY is available
+# ✅ Check if the API_KEY is set
 if not API_KEY:
     raise ValueError("API key is missing. Please set the SERPAPI_API_KEY environment variable.")
 
 def get_collaboration_data(competitors, target="Virgin Media"):
     results = []
     
+    # Iterate over each competitor to search for collaborations with Virgin Media
     for comp in competitors:
         query = f"{comp} {target} partnership OR collaboration"
         params = {
@@ -48,4 +49,3 @@ def get_collaboration_data(competitors, target="Virgin Media"):
 
     # Return results as a DataFrame
     return pd.DataFrame(results)
-