Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import pipeline
 from deepface import DeepFace
 from PIL import Image
 import io
+import re
 
 # Load Fake News Detection Model from Hugging Face
 fake_news_pipeline = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-fake-news-detection")
@@ -29,18 +30,34 @@ def verify_news(news_text):
     search_url = f"https://www.google.com/search?q={'+'.join(news_text.split())}"
     return search_url
 
-def check_video_fake_news(video_url):
-    """Uses Google Fact-Check API for video verification."""
-    api_url = f"https://factchecktools.googleapis.com/v1alpha1/claims:search?query={video_url}&key=YOUR_API_KEY"
-    response = requests.get(api_url)
+def extract_video_id(video_url):
+    """Extracts the video ID from a YouTube URL."""
+    pattern = r"(?:https?:\/\/)?(?:www\.)?(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|.*[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+    match = re.search(pattern, video_url)
+    return match.group(1) if match else None
+
+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
+
+    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}"
     
+    response = requests.get(metadata_url)
     if response.status_code == 200:
         data = response.json()
-        if 'claims' in data:
-            return "Fake", 85.0  # Fake News Detected (Dummy Value)
-        else:
-            return "Real", 92.0  # Seems Real (Dummy Value)
-    return "Unknown", 0.0  # Could Not Verify
+        if "items" in data and len(data["items"]) > 0:
+            video_details = data["items"][0]["snippet"]
+            video_title = video_details["title"]
+            video_description = video_details["description"]
+            combined_text = video_title + " " + video_description
+
+            # Classify the video metadata text
+            result, accuracy = classify_text(combined_text)
+            return result, accuracy
+    return "Unknown", 0.0
 
 # Streamlit UI
 st.set_page_config(page_title="Fake News Detector", layout="wide")
@@ -88,7 +105,7 @@ with col3:
         if not video_url.strip():
             st.warning("Please enter a valid video link.")
         else:
-            result, accuracy = check_video_fake_news(video_url)
+            result, accuracy = fetch_video_metadata(video_url)
             st.session_state["video_result"] = result
             st.session_state["video_accuracy"] = accuracy
             st.session_state["video_url"] = video_url  # Store Video URL for Display
@@ -117,5 +134,9 @@ if "image_result" in st.session_state:
 # 🔹 Video Result
 if "video_result" in st.session_state:
     st.video(st.session_state["video_url"])
-    st.info(f"🎥 **Video Analysis Result:** {st.session_state['video_result']} (Accuracy: {st.session_state['video_accuracy']}%)")
-
+    if st.session_state["video_result"] == "Fake":
+        st.error(f"❌ **This video is Fake!** (Accuracy: {st.session_state['video_accuracy']}%)")
+    elif st.session_state["video_result"] == "Real":
+        st.success(f"✅ **This video is Real!** (Accuracy: {st.session_state['video_accuracy']}%)")
+    else:
+        st.warning("⚠️ Unable to verify the authenticity of this video.")