Add sentiment analysis feature using Gemini API and update UI for sentiment display

.gitignore

@@ -4,6 +4,7 @@
 __pycache__/
 *.py[cod]
 *$py.class
+model_files/
 
 # Distribution / packaging
 .Python

app.py

@@ -5,6 +5,8 @@ import torch
 import numpy as np
 from flask import Flask, request, jsonify, render_template
 import subprocess # For calling gdown
+import google.generativeai as genai
+import time
 
 os.environ['HF_HUB_DISABLE_SYMLINKS_WARNING'] = '1'
 import torch.nn as nn
@@ -333,6 +335,57 @@ def process_input(text, t5_tokenizer, encoder_tokenizer, char_to_id, current_max
     }
 # --- END HELPER METHODS ---
 
+# --- Gemini Configuration ---
+GEMINI_API_KEY = "AIzaSyDmXVXoliNUt3Pnx4_lR2aOezLaItpIXz0"
+try:
+    genai.configure(api_key=GEMINI_API_KEY)
+except Exception as e:
+    print(f"Could not configure Gemini: {e}")
+
+def get_sentiment_from_gemini(text: str, retries: int = 3, delay: int = 5) -> str:
+    """
+    Analyzes the sentiment of a given text using the Gemini API.
+    """
+    if not text.strip():
+        return "Neutral"
+
+    try:
+        model = genai.GenerativeModel(model_name="gemini-1.5-flash")
+        
+        prompt = f"""Analyze the sentiment of the following Bengali text. Your response must be a single word: 'Positive', 'Negative', or 'Neutral'.
+Text: "{text}"
+Sentiment:"""
+
+        for attempt in range(retries):
+            try:
+                response = model.generate_content(
+                    prompt,
+                    generation_config=genai.types.GenerationConfig(
+                        candidate_count=1,
+                        max_output_tokens=5,
+                        temperature=0.1
+                    )
+                )
+                sentiment = response.text.strip().capitalize()
+                
+                if sentiment in ["Positive", "Negative", "Neutral"]:
+                    print(f"Gemini sentiment for '{text[:30]}...': {sentiment}")
+                    return sentiment
+                else:
+                    print(f"Gemini returned unexpected sentiment: '{sentiment}'. Attempt {attempt + 1}/{retries}")
+            
+            except Exception as e:
+                print(f"Error calling Gemini API on attempt {attempt + 1}/{retries}: {e}")
+
+            if attempt < retries - 1:
+                print(f"Retrying in {delay} seconds...")
+                time.sleep(delay)
+
+        print("Gemini sentiment analysis failed after all retries.")
+        return "Unknown"
+    except Exception as e:
+        print(f"An error occurred during Gemini model initialization: {e}")
+        return "Error"
 
 app = Flask(__name__)
 
@@ -412,7 +465,11 @@ def translate_text():
         translation = CFG.t5_tokenizer.decode(
             generated_ids[0], skip_special_tokens=True, clean_up_tokenization_spaces=True
         ).strip()
-        return jsonify({"translation": translation})
+        
+        # Get sentiment analysis from Gemini
+        sentiment = get_sentiment_from_gemini(translation)
+
+        return jsonify({"translation": translation, "sentiment": sentiment})
     except Exception as e:
         print(f"Error during translation: {e}")
         # import traceback

static/style.css

@@ -185,6 +185,37 @@ textarea:focus {
     line-height: 1.6;
 }
 
+/* Sentiment Badge Styles */
+#sentimentContainer {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+    font-size: 1em;
+    color: var(--text-color);
+}
+
+.sentiment-badge {
+    display: inline-block;
+    padding: 4px 12px;
+    border-radius: 16px;
+    font-size: 0.9em;
+    font-weight: 500;
+    color: white;
+    text-transform: capitalize;
+}
+
+.sentiment-positive {
+    background-color: #2e7d32;
+}
+
+.sentiment-negative {
+    background-color: #c62828;
+}
+
+.sentiment-neutral {
+    background-color: #616161;
+}
+
 /* Loading Spinner */
 .loading-spinner {
     display: flex;

templates/index.html

@@ -53,6 +53,9 @@
                 </div>
                 <div class="translation-result">
                     <p id="translationResult"></p>
+                    <div id="sentimentContainer" style="display: none; margin-top: 15px;">
+                        <strong>Sentiment:</strong> <span id="sentimentResult" class="sentiment-badge"></span>
+                    </div>
                     <div class="loading-spinner" style="display: none;">
                         <div class="spinner"></div>
                     </div>
@@ -70,6 +73,8 @@
             const copyBtn = document.getElementById('copyBtn');
             const translationResult = document.getElementById('translationResult');
             const loadingSpinner = document.querySelector('.loading-spinner');
+            const sentimentContainer = document.getElementById('sentimentContainer');
+            const sentimentResult = document.getElementById('sentimentResult');
 
             // Character counter
             userInput.addEventListener('input', function() {
@@ -80,6 +85,9 @@
             clearBtn.addEventListener('click', function() {
                 userInput.value = '';
                 translationResult.textContent = '';
+                sentimentContainer.style.display = 'none';
+                sentimentResult.textContent = '';
+                sentimentResult.className = 'sentiment-badge';
                 charCount.textContent = '0/500';
             });
 
@@ -108,6 +116,8 @@
                 loadingSpinner.style.display = 'flex';
                 translateBtn.disabled = true;
                 translationResult.style.opacity = '0.5';
+                translationResult.textContent = '';
+                sentimentContainer.style.display = 'none';
 
                 try {
                     const response = await fetch('/translate', {
@@ -120,6 +130,14 @@
                     
                     const data = await response.json();
                     translationResult.textContent = data.translation;
+
+                    if (data.sentiment && data.sentiment !== 'Error' && data.sentiment !== 'Unknown') {
+                        sentimentResult.textContent = data.sentiment;
+                        sentimentResult.className = 'sentiment-badge'; // Reset classes
+                        sentimentResult.classList.add(`sentiment-${data.sentiment.toLowerCase()}`);
+                        sentimentContainer.style.display = 'flex';
+                    }
+
                     translationResult.style.opacity = '1';
                 } catch (error) {
                     console.error('Error:', error);