Update app.py

app.py

@@ -12,8 +12,14 @@ from modules.record import DatabaseComponent  # Import DatabaseComponent
 # Load environment variables
 load_dotenv()
 
-# Initialize database and executor
-db = DatabaseComponent()
+# Initialize database (handle connection failures)
+db = None
+try:
+    db = DatabaseComponent()
+except Exception as e:
+    print(f"[ERROR] Database connection failed: {str(e)}")
+
+# Initialize thread executor
 executor = ThreadPoolExecutor(max_workers=3)
 
 # Constants for file paths and API keys
@@ -69,8 +75,9 @@ def evaluate_news(news_input):
         else:
             status = f"**{truth_percentage:.0f}% chances to be true** - This news is unlikely to be true. Proceed with caution."
 
-        # Save result in database
-        db.save_news_verification(news_text[:100], truth_score, first_citation)
+        # Save result in database if connection is available
+        if db is not None:
+            db.save_news_verification(news_text[:100], truth_score, first_citation)
 
         # Initial result
         result = f"**News:** \"{news_text[:300]}...\"\n\n"
@@ -99,6 +106,9 @@ def evaluate_news(news_input):
 
 # Function to fetch dashboard data
 def fetch_dashboard_data():
+    if db is None:
+        return "**⚠️ Database unavailable. Recent verification records cannot be displayed.**"
+
     total_news = db.get_total_news_count()
     last_10_news = db.get_last_10_news()
 
@@ -152,4 +162,4 @@ with gr.Blocks(css="""
     gr.Markdown("### **About EchoTruth**")
     gr.Markdown("EchoTruth uses AI to help users verify news authenticity in real-time.")
 
-demo.launch()
+demo.launch()