Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import random
 from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import io
@@ -8,6 +9,85 @@ from pdfminer.layout import LAParams
 from pdfminer.pdfpage import PDFPage
 from docx import Document
 
+# --- Color Themes ---
+THEMES = {
+    "midnight": {
+        "primaryColor": "#262730",
+        "backgroundColor": "#0E0F19",
+        "secondaryBackgroundColor": "#262730",
+        "textColor": "#FAFAFA",
+        "font": "sans-serif",
+        "welcomeTextColor": "#808080",
+        "tileColor": "#404040",
+    },
+    "forest": {
+        "primaryColor": "#38761d",
+        "backgroundColor": "#228B22",
+        "secondaryBackgroundColor": "#568259",
+        "textColor": "#E0FFFF",
+        "font": "sans-serif",
+        "welcomeTextColor": "#E0FFFF",
+        "tileColor": "#3D59AB",
+    },
+    "beach": {
+        "primaryColor": "#F4D03F",
+        "backgroundColor": "#A1D1E8",
+        "secondaryBackgroundColor": "#F7CAC9",
+        "textColor": "#6A5149",
+        "font": "sans-serif",
+        "welcomeTextColor": "#795548",
+        "tileColor": "#C3B091",
+    },
+    "sunset": {
+        "primaryColor": "#FFB347",
+        "backgroundColor": "#E07A5F",
+        "secondaryBackgroundColor": "#81B29A",
+        "textColor": "#F4F1DE",
+        "font": "sans-serif",
+        "welcomeTextColor": "#F2CC8F",
+        "tileColor": "#3D405B",
+    },
+}
+
+# --- Function to Apply Theme ---
+def apply_theme(theme):
+    st.markdown(f"""
+        <style>
+            body {{
+                color: {theme["textColor"]};
+                background-color: {theme["backgroundColor"]};
+                font-family: {theme["font"]};
+            }}
+            .welcome-text {{
+                color: {theme["welcomeTextColor"]};
+                font-size: 36px;
+                font-weight: bold;
+                text-align: center;
+                margin-bottom: 20px;
+            }}
+            .stSlider > div > div {{
+                color: {theme["textColor"]} !important;
+            }}
+            .stMetricLabel {{
+                color: {theme["textColor"]} !important;
+            }}
+            .stMetricValue {{
+                color: {theme["textColor"]} !important;
+            }}
+            .output-box {{
+                background-color: {theme["tileColor"]};
+                color: {theme["textColor"]};
+                padding: 10px;
+                border-radius: 5px;
+                margin-top: 20px;
+            }}
+            .stFileUploader > div > div:nth-child(1) > div > button {{
+                background-color: {theme["primaryColor"]};
+                color: {theme["textColor"]};
+            }}
+        </style>
+        """, unsafe_allow_html=True)
+
 # Functions for file processing
 def extract_text_from_pdf(pdf_file):
     resource_manager = PDFResourceManager()
@@ -103,48 +183,18 @@ def plagiarism_check(text_chunks, tokenizer, model):
 
 # Streamlit app with styling and multiple file uploads
 def main():
+    # --- Random Theme Selection ---
+    selected_theme = random.choice(list(THEMES.values()))
+    apply_theme(selected_theme)
+
     # Set page configuration and background color
     st.set_page_config(page_title="AI & Plagiarism Detection", page_icon="🔍")
-    
-    # Custom CSS for styling the app's appearance
-    st.markdown(
-        """
-        <style>
-            body {
-                background-color: black;
-                color: white;
-                font-family: 'Helvetica Neue', sans-serif;
-            }
-            .title {
-                color: white;
-                font-size: 24px;
-                font-weight: bold;
-                text-align: center;
-            }
-            .red-wine {
-                background-color: #8B0000; /* Red wine color */
-                padding: 10px;
-                border-radius: 5px;
-                margin-bottom: 20px;
-            }
-            .output-box {
-                background-color: #8B0000; /* Red wine color */
-                padding: 10px;
-                border-radius: 5px;
-                margin-top: 20px;
-            }
-            .stSlider > div > div {
-                color: white !important; /* Slider text color */
-            }
-        </style>
-        """,
-        unsafe_allow_html=True,
-    )
 
     # Title with slider effect
-    slider_value = st.slider("AI Plagiarism Detection", min_value=0, max_value=100, value=50)
+    slider_value = st.slider("AI Plagiarism Detection Tool", min_value=0, max_value=100, value=50)
 
-    st.markdown("<h1 class='title'>Welcome to AI & Plagiarism Detection</h1>", unsafe_allow_html=True)
+    # Welcome style text
+    st.markdown("<h1 class='welcome-text'>Welcome to AI & Plagiarism Detection</h1>", unsafe_allow_html=True)
 
     # Load models
     ai_detection_model = load_ai_detection_model()
@@ -187,20 +237,21 @@ def main():
                     else:
                         ai_percentage_avg = None
                         human_percentage = None
-                
-                    # Plagiarism Check
-                    if tokenizer and plagiarism_model:
-                        plagiarism_percentage = plagiarism_check(text_chunks, tokenizer, plagiarism_model)
-                    else:
-                        plagiarism_percentage = None
 
-                    # Display Results in separate tiles for each file processed.
+                # Plagiarism Check
+                if tokenizer and plagiarism_model:
+                    plagiarism_percentage = plagiarism_check(text_chunks, tokenizer, model)
+                else:
+                    plagiarism_percentage = None
+
+                # --- Tiled Output ---
+                with st.container():
                     st.markdown(f"<div class='output-box'><h3>{uploaded_file.name}</h3></div>", unsafe_allow_html=True)
 
                     col1, col2 = st.columns(2)
 
                     with col1:
-                        st.markdown("<div class='red-wine'><h4>AI Detection:</h4></div>", unsafe_allow_html=True)
+                        st.markdown("<div class='output-box'><h4>AI Detection:</h4></div>", unsafe_allow_html=True)
                         if ai_percentage_avg is not None:
                             st.metric(label="AI Content", value=f"{ai_percentage_avg:.2f}%", delta="AI Generated")
                             st.metric(label="Human Written", value=f"{human_percentage:.2f}%", delta="Humanized Text")
@@ -208,7 +259,7 @@ def main():
                             st.write("AI Detection not available")
 
                     with col2:
-                        st.markdown("<div class='red-wine'><h4>Plagiarism Detection:</h4></div>", unsafe_allow_html=True)
+                        st.markdown("<div class='output-box'><h4>Plagiarism Detection:</h4></div>", unsafe_allow_html=True)
                         if plagiarism_percentage is not None:
                             st.metric(label="Plagiarism", value=f"{plagiarism_percentage:.2f}%", delta="Plagiarized" if plagiarism_percentage > 0 else "Original")
                         else: