Update app.py

app.py

@@ -71,22 +71,23 @@ MOTHERDUCK_TOKEN = st.secrets["MOTHERDUCK_TOKEN"]
 def get_data():
     try:
         # Connect to MotherDuck using the token
-        # This creates an in-memory connection that links to the cloud DB
         con = duckdb.connect(f'md:?token={MOTHERDUCK_TOKEN}')
         
+        # Updated query: Removed 'id' based on your schema
         query = """
-            SELECT id, entity, entity_score, domain, title_score, overall_score, created_at
+            SELECT entity, entity_score, domain, title_score, overall_score, created_at
             FROM sentiment_analysis
         """
         
-        # Execute query and fetch result as a Pandas DataFrame
         df = con.sql(query).df()
         
-        # Check if created_at needs conversion (if it's a timestamp/int)
-        # If MotherDuck returns datetime objects, this line might need adjustment,
-        # but keeping original logic assuming Unix timestamp storage.
-        if not pd.api.types.is_datetime64_any_dtype(df["created_at"]):
+        # Robust date handling:
+        # If it's numeric (Unix timestamp), convert using unit='s'.
+        # If it's already a TIMESTAMP/Datetime object (MotherDuck native), ensure it's standard pandas datetime.
+        if pd.api.types.is_numeric_dtype(df["created_at"]):
              df["created_at"] = pd.to_datetime(df["created_at"], unit="s")
+        else:
+             df["created_at"] = pd.to_datetime(df["created_at"])
              
         return df
     except Exception as e: