Update db.py

db.py

@@ -1,40 +1,55 @@
 import sqlite3
-import pandas as pd
+from datetime import datetime
 
-DB_FILE = "trends.db"
+DB_FILE = 'data/trends.db'
 
 def init_db():
     conn = sqlite3.connect(DB_FILE)
     c = conn.cursor()
-    c.execute('''CREATE TABLE IF NOT EXISTS trends_history (
-                    id INTEGER PRIMARY KEY AUTOINCREMENT,
-                    country TEXT,
-                    topic TEXT,
-                    sentiment REAL,
-                    date TIMESTAMP DEFAULT CURRENT_TIMESTAMP
-                )''')
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS trends_history (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            country TEXT,
+            topic TEXT,
+            date TIMESTAMP,
+            sentiment REAL
+        )
+    ''')
     conn.commit()
     conn.close()
 
-
-def insert_trends(data):
+def insert_trends(country, trends):
     conn = sqlite3.connect(DB_FILE)
-    df = pd.DataFrame(data)
-    df.to_sql('trends_history', conn, if_exists='append', index=False)
+    c = conn.cursor()
+    for trend in trends:
+        c.execute('''
+            INSERT INTO trends_history (country, topic, date, sentiment)
+            VALUES (?, ?, ?, ?)
+        ''', (country, trend['topic'], datetime.now(), trend['sentiment']))
+    conn.commit()
     conn.close()
 
-
-def get_country_trends(country):
+def fetch_trends_by_country(country, limit=10):
     conn = sqlite3.connect(DB_FILE)
-    query = "SELECT country, topic, sentiment, date FROM trends_history WHERE country=? ORDER BY date DESC LIMIT 100"
-    df = pd.read_sql_query(query, conn, params=(country,))
+    c = conn.cursor()
+    c.execute('''
+        SELECT topic, sentiment, date FROM trends_history
+        WHERE country = ?
+        ORDER BY date DESC
+        LIMIT ?
+    ''', (country, limit))
+    rows = c.fetchall()
     conn.close()
-    return df
-
+    return rows
 
-def get_trend_history():
+def fetch_global_trends(limit=10):
     conn = sqlite3.connect(DB_FILE)
-    query = "SELECT * FROM trends_history ORDER BY date DESC LIMIT 500"
-    df = pd.read_sql_query(query, conn)
+    c = conn.cursor()
+    c.execute('''
+        SELECT country, topic, sentiment, date FROM trends_history
+        ORDER BY date DESC
+        LIMIT ?
+    ''', (limit,))
+    rows = c.fetchall()
     conn.close()
-    return df
+    return rows