Update app.py

app.py

@@ -3,13 +3,13 @@ import pandas as pd
 import io
 import matplotlib.pyplot as plt
 from script import getSearchResult, getClustersWithGraph, compare_clusters
-import os
 
 app = marimo.App()
 
 @app.cell
 def _(mo):
     csv_upload = mo.ui.file(label="Upload Keyword CSV (1 column)")
+    import os
     api_key = os.environ.get("GOOGLE_API_KEY", "")
     cse_id = os.environ.get("GOOGLE_CSE_ID", "")
     country = mo.ui.text(label="Country Code (e.g. UK)", value="UK")
@@ -69,4 +69,10 @@ def _(movement):
     if movement is not None:
         movement.sort_values(by="searchTerms", inplace=True)
         movement.reset_index(drop=True, inplace=True)
-        return movement
+        return movement is not None:
+        movement.sort_values(by="searchTerms", inplace=True)
+        movement.reset_index(drop=True, inplace=True)
+        return movement is not None:
+        movement.sort_values(by="searchTerms", inplace=True)
+        movement.reset_index(drop=True, inplace=True)
+        movement