Update app.py

app.py

@@ -3,11 +3,14 @@ import requests
 from transformers import pipeline
 from PIL import Image
 import torchvision.transforms as transforms
+import torch
+import cv2
+import numpy as np
 
-# Load a more accurate fake news detection model
+# Load Fake News Detection Model
 fake_news_pipeline = pipeline("text-classification", model="roberta-base-openai-detector")
 
-# Function to classify news text
+# Function to classify text news
 def classify_text(news_text):
     result = fake_news_pipeline(news_text)[0]
     label = result['label'].lower()
@@ -19,11 +22,35 @@ def verify_news(news_text):
     search_url = f"https://toolbox.google.com/factcheck/explorer/search/{'+'.join(news_text.split())}"
     return search_url
 
-# Function to analyze images (Dummy function for now)
+# Function to analyze images for fake news
 def analyze_image(image):
     transform = transforms.Compose([transforms.Resize((224, 224)), transforms.ToTensor()])
     image_tensor = transform(image).unsqueeze(0)
-    return "Image analysis feature coming soon!"
+
+    # Convert to OpenCV format
+    image_cv = np.array(image)
+    image_cv = cv2.cvtColor(image_cv, cv2.COLOR_RGB2BGR)
+
+    # Apply Fake Image Detection (Basic Example)
+    edges = cv2.Canny(image_cv, 100, 200)
+    edge_percentage = np.sum(edges) / (image_cv.shape[0] * image_cv.shape[1])
+
+    if edge_percentage > 0.1:
+        return "❌ This image might be **manipulated** or **deepfake**!"
+    else:
+        return "✅ This image appears to be **authentic**."
+
+# Function to analyze video metadata
+def analyze_video(video_url):
+    api_url = f"https://noembed.com/embed?url={video_url}"
+    response = requests.get(api_url).json()
+
+    if "title" in response:
+        title = response["title"]
+        source = response["provider_name"]
+        return f"🎥 Video Title: {title}\n🔗 Source: {source}"
+    else:
+        return "❌ Unable to fetch video details! Check if the link is correct."
 
 # Streamlit UI
 st.set_page_config(page_title="Fake News Detector", layout="wide")
@@ -61,7 +88,8 @@ with col2:
     if uploaded_image and st.button("Analyze Image"):
         image = Image.open(uploaded_image)
         st.image(image, caption="Uploaded Image", use_column_width=True)
-        st.info(analyze_image(image))
+        result = analyze_image(image)
+        st.info(result)
 
 # Video Link Section
 with col3:
@@ -72,5 +100,5 @@ with col3:
             st.warning("Please enter a valid video link.")
         else:
             st.video(video_url)
-            st.info("Video analysis feature coming soon!")
-
+            result = analyze_video(video_url)
+            st.info(result)