Add Telegram integration and SQLite database for sentiment analysis results

.env

@@ -0,0 +1,2 @@
+TELEGRAM_BOT_TOKEN=<your-telegram-bot-token>
+TELEGRAM_CHAT_ID=<your-chat-id>

app.py

@@ -1,5 +1,12 @@
 import gradio as gr
 from transformers import pipeline
+import sqlite3
+import requests
+from dotenv import load_dotenv
+import os
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Load the Sentiment Analysis pipeline...
 classifier = pipeline(
@@ -7,6 +14,38 @@ classifier = pipeline(
     model="distilbert-base-uncased-finetuned-sst-2-english"
 )
 
+# Initialize SQLite database connection
+conn = sqlite3.connect("sentiment_analysis.db")
+cursor = conn.cursor()
+
+# Create a table to store input and output if it doesn't exist
+cursor.execute("""
+CREATE TABLE IF NOT EXISTS SentimentAnalysis (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    input_text TEXT NOT NULL,
+    output_text TEXT NOT NULL,
+    confidence_score REAL NOT NULL
+)
+""")
+conn.commit()
+
+# Define your Telegram bot token and chat ID
+TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID")
+
+# Function to send messages to the Telegram bot
+def send_to_telegram(message):
+    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
+    payload = {
+        "chat_id": TELEGRAM_CHAT_ID,
+        "text": message,
+        "parse_mode": "Markdown"
+    }
+    try:
+        requests.post(url, data=payload)
+    except requests.exceptions.RequestException as e:
+        print(f"Failed to send message to Telegram: {e}")
+
 # Define the prediction function...
 def sentiment_predictor(text):
     if not text:
@@ -17,6 +56,22 @@ def sentiment_predictor(text):
     score = result['score']
 
     output_text = f"Predicted Sentiment: **{label}**"
+
+    # Save input and output to the database
+    cursor.execute(
+        "INSERT INTO SentimentAnalysis (input_text, output_text, confidence_score) VALUES (?, ?, ?)",
+        (text, output_text, score)
+    )
+    conn.commit()
+
+    # Send input and output to the Telegram bot
+    message = (
+        f"*New Sentiment Analysis Result:*\n"
+        f"*Input:* {text}\n"
+        f"*Output:* {output_text}\n"
+        f"*Confidence Score:* {score:.2f}"
+    )
+    send_to_telegram(message)
     
     return output_text, score
 

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 transformers
-torch
+torch
+python-dotenv