Update app.py

app.py

@@ -5,6 +5,7 @@ from deepface import DeepFace
 from PIL import Image
 import io
 import re
+import base64
 
 # Load Fake News Detection Model from Hugging Face
 fake_news_pipeline = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-fake-news-detection")
@@ -17,18 +18,34 @@ def classify_text(news_text):
     return ("Fake" if label == "fake" else "Real"), round(score, 2)
 
 def analyze_image(image):
-    """Analyzes image using DeepFace and provides a result."""
+    """Analyzes image using DeepFace and Google Reverse Image Search."""
     try:
         analysis = DeepFace.analyze(image, actions=['emotion'])
         dominant_emotion = analysis[0]['dominant_emotion']
-        return f"Analysis Complete - Dominant Emotion: {dominant_emotion}", 90.0  # Dummy Accuracy
+        reverse_search_url = reverse_image_search(image)
+        return f"Analysis: {dominant_emotion}", 90.0, reverse_search_url  # Dummy Accuracy
     except Exception as e:
-        return f"Error: {str(e)}", 0.0
+        return f"Error: {str(e)}", 0.0, None
+
+def reverse_image_search(image):
+    """Creates a Google Reverse Image Search link for verification."""
+    buffered = io.BytesIO()
+    image.save(buffered, format="PNG")
+    encoded_img = base64.b64encode(buffered.getvalue()).decode()
+    return f"https://www.google.com/searchbyimage?image_url=data:image/png;base64,{encoded_img}"
 
 def verify_news(news_text):
-    """Generates a Google search link for verification."""
-    search_url = f"https://www.google.com/search?q={'+'.join(news_text.split())}"
-    return search_url
+    """Searches trusted fact-checking websites for news verification."""
+    sources = [
+        ("BBC News", "https://www.bbc.com/news"),
+        ("CNN", "https://www.cnn.com"),
+        ("Reuters", "https://www.reuters.com"),
+        ("FactCheck.org", "https://www.factcheck.org"),
+        ("Snopes", "https://www.snopes.com"),
+        ("PolitiFact", "https://www.politifact.com"),
+        ("Google Search", f"https://www.google.com/search?q={'+'.join(news_text.split())}")
+    ]
+    return sources
 
 def extract_video_id(video_url):
     """Extracts the video ID from a YouTube URL."""
@@ -40,7 +57,7 @@ def fetch_video_metadata(video_url):
     """Fetches video metadata and runs Fake News detection on it."""
     video_id = extract_video_id(video_url)
     if not video_id:
-        return "Invalid Video URL", 0.0
+        return "Invalid Video URL", 0.0, None
 
     api_key = "YOUR_YOUTUBE_API_KEY"  # Replace with a valid YouTube API Key
     metadata_url = f"https://www.googleapis.com/youtube/v3/videos?id={video_id}&part=snippet&key={api_key}"
@@ -56,8 +73,9 @@ def fetch_video_metadata(video_url):
 
             # Classify the video metadata text
             result, accuracy = classify_text(combined_text)
-            return result, accuracy
-    return "Unknown", 0.0
+            verification_links = verify_news(video_title)
+            return result, accuracy, verification_links
+    return "Unknown", 0.0, None
 
 # Streamlit UI
 st.set_page_config(page_title="Fake News Detector", layout="wide")
@@ -79,8 +97,10 @@ with col1:
             st.warning("Please enter some text.")
         else:
             result, accuracy = classify_text(news_text)
+            verification_links = verify_news(news_text)
             st.session_state["text_result"] = result
             st.session_state["text_accuracy"] = accuracy
+            st.session_state["text_verification"] = verification_links
 
 # 🔹 Image Upload Section
 with col2:
@@ -90,9 +110,10 @@ with col2:
     
     if uploaded_image and analyze_image_clicked:
         image = Image.open(uploaded_image)
-        result, accuracy = analyze_image(image)
+        result, accuracy, reverse_search_url = analyze_image(image)
         st.session_state["image_result"] = result
         st.session_state["image_accuracy"] = accuracy
+        st.session_state["image_search_url"] = reverse_search_url
         st.session_state["news_image"] = image  # Store Image for Display
 
 # 🔹 Video Link Section
@@ -105,9 +126,10 @@ with col3:
         if not video_url.strip():
             st.warning("Please enter a valid video link.")
         else:
-            result, accuracy = fetch_video_metadata(video_url)
+            result, accuracy, verification_links = fetch_video_metadata(video_url)
             st.session_state["video_result"] = result
             st.session_state["video_accuracy"] = accuracy
+            st.session_state["video_verification"] = verification_links
             st.session_state["video_url"] = video_url  # Store Video URL for Display
 
 # 🔹 Results Section
@@ -123,13 +145,15 @@ if "text_result" in st.session_state:
     else:
         st.success(f"✅ This news is **Real**! (Accuracy: {accuracy}%)", icon="✅")
     
-    verification_link = verify_news(news_text)
-    st.markdown(f"[🔎 Verify on Google]({verification_link})")
+    st.subheader("🔎 Trusted Fact-Checking Sources")
+    for name, link in st.session_state["text_verification"]:
+        st.markdown(f"[🔗 {name}]({link})")
 
 # 🔹 Image Result
 if "image_result" in st.session_state:
     st.image(st.session_state["news_image"], caption="Uploaded Image", use_column_width=True)
     st.info(f"🖼️ **Image Analysis Result:** {st.session_state['image_result']} (Accuracy: {st.session_state['image_accuracy']}%)")
+    st.markdown(f"[🔎 Google Reverse Image Search]({st.session_state['image_search_url']})")
 
 # 🔹 Video Result
 if "video_result" in st.session_state:
@@ -140,3 +164,7 @@ if "video_result" in st.session_state:
         st.success(f"✅ **This video is Real!** (Accuracy: {st.session_state['video_accuracy']}%)")
     else:
         st.warning("⚠️ Unable to verify the authenticity of this video.")
+    
+    st.subheader("🔎 Trusted Fact-Checking Sources")
+    for name, link in st.session_state["video_verification"]:
+        st.markdown(f"[🔗 {name}]({link})")