Update app.py

app.py

@@ -51,54 +51,77 @@ import tempfile
 import os
 from PIL import Image
 
-def compress_image(image, quality=20):
-    img = Image.open(image)
-    img = img.convert("RGB")
+st.set_page_config(page_title="Fake News & Deepfake Detection", layout="wide")
+
+# 👉 Image Compression Function
+def compress_image(image, quality=20, max_size=(500, 500)):
+    img = Image.open(image).convert("RGB")
+    img.thumbnail(max_size)  # Resize while keeping aspect ratio
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
     img.save(temp_file.name, "JPEG", quality=quality)
     return temp_file.name
 
+# 👉 Video Compression Function
 def compress_video(video):
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-    cap = cv2.VideoCapture(video)
+
+    # ✅ Save video to temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video:
+        temp_video.write(video.read())
+        video_path = temp_video.name
+
+    cap = cv2.VideoCapture(video_path)
+    
+    if not cap.isOpened():
+        st.error("❌ Error: Unable to read video!")
+        return None
+
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    frame_width = int(cap.get(3) // 2)
-    frame_height = int(cap.get(4) // 2)
+
+    # ✅ Reduce resolution to 480p
+    frame_width = 640  
+    frame_height = 480  
     out = cv2.VideoWriter(temp_file.name, fourcc, 20.0, (frame_width, frame_height))
-    
+
     while cap.isOpened():
         ret, frame = cap.read()
         if not ret:
             break
         frame = cv2.resize(frame, (frame_width, frame_height))
         out.write(frame)
-    
+
     cap.release()
     out.release()
+    
     return temp_file.name
 
+# 👉 Streamlit UI
 st.title("🕵️‍♂️ Fake News & Deepfake Detection Tool")
 
-st.sidebar.header("Upload your file")
-option = st.sidebar.radio("Select file type", ["Image", "Video", "Text"])
+st.sidebar.header("📂 Upload Your File")
+option = st.sidebar.radio("Select File Type", ["📷 Image", "📹 Video", "📝 Text"])
 
-if option == "Image":
-    uploaded_file = st.sidebar.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-    if uploaded_file is not None:
+# 👉 Image Upload & Compression
+if option == "📷 Image":
+    uploaded_file = st.sidebar.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
+    if uploaded_file:
         compressed_path = compress_image(uploaded_file)
         image = Image.open(compressed_path)
-        st.image(image, caption="Compressed Image", use_column_width=True)
+        st.image(image, caption="🖼️ Compressed Image", use_column_width=True)
         st.success("✅ Image uploaded and compressed successfully!")
 
-elif option == "Video":
-    uploaded_file = st.sidebar.file_uploader("Upload a video", type=["mp4", "avi", "mov"])
-    if uploaded_file is not None:
+# 👉 Video Upload & Compression
+elif option == "📹 Video":
+    uploaded_file = st.sidebar.file_uploader("Upload a Video", type=["mp4", "avi", "mov"])
+    if uploaded_file:
         compressed_path = compress_video(uploaded_file)
-        st.video(compressed_path)
-        st.success("✅ Video uploaded and compressed successfully!")
+        if compressed_path:
+            st.video(compressed_path)
+            st.success("✅ Video uploaded and compressed successfully!")
 
-elif option == "Text":
+# 👉 Text Input for Fake News Detection
+elif option == "📝 Text":
     text_input = st.text_area("Enter your text for analysis")
     if text_input:
         st.write("🔍 Fake news detection processing...")
-        st.success("✅ Text analysis completed!")
+        st.success("✅ Text analysis completed!")