Update app.py

app.py

@@ -5,9 +5,9 @@ import plotly.graph_objects as go
 import streamlit as st
 from textblob import TextBlob
 import json
+import google.generativeai as genai
 import os
 import concurrent.futures
-from groq import Groq
 import textstat
 import trafilatura
 import requests
@@ -23,8 +23,11 @@ URL_A = "https://www.foxnews.com/live-news/trump-iran-israel-war-updates-march-3
 URL_B = "https://edition.cnn.com/2026/03/30/world/live-news/iran-war-us-israel-trump"
 
 # Initialize the Hugging Face Client
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
-client = Groq(api_key=GROQ_API_KEY)
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
+if GEMINI_API_KEY:
+    genai.configure(api_key=GEMINI_API_KEY)
+
+ai_model = genai.GenerativeModel('gemini-1.5-pro')
 
 def _truncate_to_words(text: str, limit: int) -> str:
     """Truncates text by word count."""
@@ -64,14 +67,14 @@ def analyze_article(text: str) -> dict:
     """
     
     messages = [{"role": "user", "content": prompt}]
-    response = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
-        messages=messages,
-        max_tokens=300,
-        temperature=0.1,
-        response_format={"type": "json_object"}
+    response = ai_model.generate_content(
+        prompt,
+        generation_config=genai.GenerationConfig(
+            response_mime_type="application/json",
+            temperature=0.1,
+        )
     )
-    llm_data = json.loads(response.choices[0].message.content)
+    llm_data = json.loads(response.text)
 
     subjectivity_score = TextBlob(safe_text).sentiment.subjectivity
     raw_reading_ease = textstat.flesch_reading_ease(safe_text)
@@ -215,23 +218,22 @@ def check_contradiction(text_a: str, text_b: str) -> dict:
     Text 2: "{safe_b}"
     """
     messages = [{"role": "user", "content": prompt}]
-    response = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",
-        messages=messages,
-        max_tokens=100,
-        temperature=0.1,
-        response_format={"type": "json_object"}
+    response = ai_model.generate_content(
+        prompt,
+        generation_config=genai.GenerationConfig(
+            response_mime_type="application/json",
+            temperature=0.1,
+        )
     )
-    
-    result = json.loads(response.choices[0].message.content)
+    result = json.loads(response.text)
     return {"relationship": result.get("relationship", "NEUTRAL"), "confidence": result.get("confidence", 0.0)}
     
 
 # USER INTERFACE
 st.set_page_config(page_title="FrameVis | Media Framing", layout="wide")
 
-if not GROQ_API_KEY:
-    st.warning("Groq API Token Missing.")
+if not GEMINI_API_KEY:
+    st.warning("Gemini API Token Missing.")
     st.stop()
 
 st.markdown("""