Update app.py

app.py

@@ -4,7 +4,6 @@ from transformers import pipeline
 from PIL import Image
 import torch
 import torchvision.transforms as transforms
-import io
 
 # Load Fake News Detection Model from Hugging Face
 fake_news_pipeline = pipeline("text-classification", model="mrm8488/bert-tiny-finetuned-fake-news-detection")
@@ -28,43 +27,50 @@ def verify_news(news_text):
 st.set_page_config(page_title="Fake News Detector", layout="wide")
 st.title("📰 Fake News Detector")
 
-# Ensure session state variables are initialized before modifying them
-for key in ["analyze_text", "result_text", "accuracy_text", "analyze_image", "analyze_video", "news_text", "news_image", "video_url"]:
-    if key not in st.session_state:
-        st.session_state[key] = None
+# Dividing Input Section
+col1, col2, col3 = st.columns(3)
 
-# Input Section
-st.header("1️⃣ Text Input")
-news_text = st.text_area("Enter the news content to check:", height=200)
-analyze_text_clicked = st.button("Analyze Text")
+# Text Input Section
+with col1:
+    st.subheader("📄 Enter Text")
+    news_text = st.text_area("Enter the news content to check:", height=150)
+    analyze_text_clicked = st.button("Analyze News")
+    if analyze_text_clicked:
+        if not news_text.strip():
+            st.warning("Please enter some text.")
+        else:
+            result, accuracy = classify_text(news_text)
+            st.session_state["news_text"] = news_text
+            st.session_state["result_text"] = result
+            st.session_state["accuracy_text"] = accuracy
 
-if analyze_text_clicked and news_text.strip():
-    result, accuracy = classify_text(news_text)
-    st.session_state["news_text"] = news_text
-    st.session_state["analyze_text"] = True
-    st.session_state["result_text"] = result
-    st.session_state["accuracy_text"] = accuracy
+# Image Upload Section
+with col2:
+    st.subheader("🖼️ Upload Image")
+    uploaded_image = st.file_uploader("Upload a news image", type=["jpg", "png", "jpeg"])
+    analyze_image_clicked = st.button("Analyze Image")
+    if uploaded_image and analyze_image_clicked:
+        image = Image.open(uploaded_image)
+        st.session_state["news_image"] = image
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.info(analyze_image(image))
 
-st.header("2️⃣ Image Upload")
-uploaded_image = st.file_uploader("Upload a news image", type=["jpg", "png", "jpeg"])
-analyze_image_clicked = st.button("Analyze Image")
-
-if uploaded_image and analyze_image_clicked:
-    image = Image.open(uploaded_image)
-    st.session_state["news_image"] = image
-    st.session_state["analyze_image"] = True
-
-st.header("3️⃣ Video Link Input")
-video_url = st.text_input("Enter the video link:")
-analyze_video_clicked = st.button("Analyze Video")
-
-if analyze_video_clicked and video_url.strip():
-    st.session_state["video_url"] = video_url
-    st.session_state["analyze_video"] = True
+# Video Link Input Section
+with col3:
+    st.subheader("🎥 Enter Video Link")
+    video_url = st.text_input("Enter the video link:")
+    analyze_video_clicked = st.button("Analyze Video")
+    if analyze_video_clicked:
+        if not video_url.strip():
+            st.warning("Please enter a valid video link.")
+        else:
+            st.session_state["video_url"] = video_url
+            st.video(video_url)
+            st.info("Video analysis feature coming soon!")
 
 # Results Section
 st.subheader("📊 Analysis Results")
-if st.session_state.get("analyze_text", False):
+if "result_text" in st.session_state:
     result = st.session_state.get("result_text")
     accuracy = st.session_state.get("accuracy_text")
     verification_link = verify_news(st.session_state.get("news_text"))
@@ -88,13 +94,3 @@ if st.session_state.get("analyze_text", False):
     for link in sources:
         st.markdown(f"[🔗 {link}]({link})")
     st.markdown(f"[🔎 Verify on Google]({verification_link})")
-
-if st.session_state.get("analyze_image", False):
-    image = st.session_state.get("news_image")
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-    st.info(analyze_image(image))
-
-if st.session_state.get("analyze_video", False):
-    video_url = st.session_state.get("video_url", "")
-    st.video(video_url)
-    st.info("Video analysis feature coming soon!")