Update main.js

main.js

@@ -72,3 +72,77 @@ setInterval(fetchTweets, 300000);
 // Initial fetch
 fetchTweets();
 
+let tweets = [];
+
+function updateStats() {
+    document.getElementById('active-disasters').textContent = 
+        tweets.filter(t => t.isDisaster).length;
+    document.getElementById('urgent-cases').textContent = 
+        tweets.filter(t => t.sentiment === 'Urgent').length;
+    document.getElementById('total-tweets').textContent = tweets.length;
+}
+
+function getSentimentClass(sentiment) {
+    switch(sentiment) {
+        case 'Urgent': return 'sentiment-urgent';
+        case 'Neutral': return 'sentiment-neutral';
+        case 'Not Urgent': return 'sentiment-not-urgent';
+        default: return '';
+    }
+}
+
+function formatTimestamp(timestamp) {
+    return new Date(timestamp).toLocaleTimeString();
+}
+
+function renderTweets() {
+    const feed = document.getElementById('tweet-feed');
+    if (tweets.length === 0) {
+        feed.innerHTML = '<p>No new tweets available. This may be due to rate limiting or no recent disaster-related tweets.</p>';
+    } else {
+        feed.innerHTML = tweets.map(tweet => `
+            <div class="tweet">
+                <div class="tweet-header">
+                    <strong>Disaster Alert</strong>
+                    <span class="sentiment-badge ${getSentimentClass(tweet.sentiment)}">
+                        ${tweet.sentiment.toUpperCase()}
+                    </span>
+                </div>
+                <div class="tweet-content">
+                    ${tweet.text}
+                </div>
+                <div class="tweet-footer">
+                    <div class="location">
+                        📍 ${tweet.location}
+                    </div>
+                    <div style="margin-left: auto">
+                        ${formatTimestamp(tweet.timestamp)}
+                    </div>
+                </div>
+            </div>
+        `).join('');
+    }
+}
+
+function fetchTweets() {
+    fetch('/api/predict')
+        .then(response => response.json())
+        .then(data => {
+            if (data.error) {
+                console.error('Error:', data.error);
+                return;
+            }
+            const newTweets = JSON.parse(data.data);
+            tweets = [...newTweets, ...tweets].slice(0, 100);  // Keep only the latest 100 tweets
+            updateStats();
+            renderTweets();
+        })
+        .catch(error => console.error('Error:', error));
+}
+
+// Fetch tweets every 5 minutes (300000 ms) instead of every 30 seconds
+setInterval(fetchTweets, 300000);
+
+// Initial fetch
+fetchTweets();
+