Upload 2 files

modules/record.py

@@ -0,0 +1,55 @@
+from dotenv import load_dotenv
+import pymysql
+import os
+
+# Load environment variables
+load_dotenv()
+
+class DatabaseComponent:
+    def __init__(self):
+        # Establish the connection
+        self.connection = pymysql.connect(
+            host=os.getenv("DB_HOST"),
+            user=os.getenv("DB_USER"),
+            password=os.getenv("DB_PASSWORD"),
+            port=int(os.getenv("DB_PORT")),
+            database=os.getenv("DB_NAME"),
+            ssl={'ca': None}  # Disable SSL for Windows
+        )
+
+    def save_news_verification(self, title, score, citation):
+        """
+        Saves a news verification result to the database.
+        """
+        with self.connection.cursor() as cursor:
+            sql = """
+            INSERT INTO news_verifications (article_title, truthfulness_score, citation)
+            VALUES (%s, %s, %s)
+            """
+            cursor.execute(sql, (title, score, citation))
+            self.connection.commit()
+
+    def get_total_news_count(self):
+        """
+        Fetches the total number of news articles checked.
+        """
+        with self.connection.cursor() as cursor:
+            sql = "SELECT COUNT(*) FROM news_verifications"
+            cursor.execute(sql)
+            result = cursor.fetchone()
+            return result[0] if result else 0
+
+    def get_last_10_news(self):
+        """
+        Fetches the last 10 saved news verification entries.
+        """
+        with self.connection.cursor() as cursor:
+            sql = """
+            SELECT article_title, truthfulness_score, citation, created_at 
+            FROM news_verifications
+            ORDER BY created_at DESC
+            LIMIT 10
+            """
+            cursor.execute(sql)
+            result = cursor.fetchall()
+            return [{"title": row[0], "score": row[1], "citation": row[2], "timestamp": row[3]} for row in result]

modules/utils.py

@@ -0,0 +1,9 @@
+import re
+
+def clean_text(text):
+    """Removes unnecessary spaces, special characters, and formats text properly."""
+    return re.sub(r'\s+', ' ', text).strip()
+
+def normalize_score(score):
+    """Ensures the score is within the range 0 to 1."""
+    return max(0, min(score, 1))