Update app.py

app.py

@@ -12,14 +12,44 @@ from docx import Document
 # --- Streamlit Page Configuration ---
 st.set_page_config(page_title="AI & Plagiarism Detection", page_icon="🔍")
 
-# --- ChatGPT Theme ---
-CHATGPT_THEME = {
-    "backgroundColor": "#343541",
-    "textColor": "#D1D5DB",
-    "tileColor": "#444654",
-    "inputAreaColor": "#40414F",
-    "accentColor": "#10a37f",
-    "font": "sans-serif",
+# --- Color Themes ---
+THEMES = {
+    "cyberpunk": {
+        "primaryColor": "#9C27B0",
+        "backgroundColor": "#1E1E1E",
+        "tileColor": "#303030",
+        "textColor": "#FFFFFF",
+        "accentColor": "#FF4081",
+        "font": "monospace",
+        "welcomeTextColor": "#FFD700"
+    },
+    "retro": {
+        "primaryColor": "#FF5733",
+        "backgroundColor": "#F0F0F0",
+        "tileColor": "#E0E0E0",
+        "textColor": "#333333",
+        "accentColor": "#64B5F6",
+        "font": "sans-serif",
+        "welcomeTextColor": "#4CAF50"
+    },
+    "darkmode": {
+        "primaryColor": "#2979FF",
+        "backgroundColor": "#212121",
+        "tileColor": "#424242",
+        "textColor": "#FFFFFF",
+        "accentColor": "#FFAB40",
+        "font": "sans-serif",
+        "welcomeTextColor": "#BBDEFB"
+    },
+    "bluelagoon": {
+        "primaryColor": "#03A9F4",
+        "backgroundColor": "#E1F5FE",
+        "tileColor": "#B3E5FC",
+        "textColor": "#333333",
+        "accentColor": "#FFCDD2",
+        "font": "sans-serif",
+        "welcomeTextColor": "#0277BD"
+    }
 }
 
 # --- Function to Apply Theme ---
@@ -32,7 +62,7 @@ def apply_theme(theme):
                 font-family: {theme["font"]};
             }}
             .welcome-text {{
-                color: {theme["textColor"]};
+                color: {theme["welcomeTextColor"]};
                 font-size: 36px;
                 font-weight: bold;
                 text-align: center;
@@ -46,12 +76,27 @@ def apply_theme(theme):
                 margin-top: 20px;
             }}
             .stTextArea textarea {{
-                background-color: {theme["inputAreaColor"]};
+                background-color: {theme["backgroundColor"]};
                 color: {theme["textColor"]};
+                border: 1px solid {theme["primaryColor"]};
+                border-radius: 5px;
             }}
             .stFileUploader > div > div:nth-child(1) > div > button {{
-                background-color: {theme["accentColor"]};
-                color: {theme["backgroundColor"]};
+                background-color: {theme["primaryColor"]};
+                color: {theme["textColor"]};
+                border-radius: 5px;
+            }}
+            .stMetricLabel {{
+                color: {theme["textColor"]} !important;
+            }}
+            .stMetricValue {{
+                color: {theme["textColor"]} !important;
+            }}
+            .streamlit-expanderHeader {{
+                color: {theme["textColor"]};
+            }}
+            .streamlit-expanderContent {{
+                color: {theme["textColor"]};
             }}
         </style>
     """, unsafe_allow_html=True)
@@ -145,15 +190,16 @@ def plagiarism_check(text_chunks, tokenizer, model):
 
 # --- Main Function ---
 def main():
-    # --- Apply ChatGPT Theme ---
-    apply_theme(CHATGPT_THEME)
+    # --- Theme Selection ---
+    theme_name = st.sidebar.selectbox("Select Theme", list(THEMES.keys()), index=0)
+    selected_theme = THEMES[theme_name]
+    apply_theme(selected_theme)
 
     # --- Title and Welcome ---
-    st.markdown("<h1 class='welcome-text'>AI & Plagiarism Detection</h1>", unsafe_allow_html=True)
+    st.markdown(f"<h1 class='welcome-text'>AI & Plagiarism Detection</h1>", unsafe_allow_html=True)
 
-    # --- Combined Input Area ---
-    input_text = st.text_area("Enter text here or attach a document:", height=200)
-    
+    # --- Input Area: Text Area and File Upload ---
+    input_text = st.text_area("Enter text or attach a document:", height=200)
     uploaded_files = st.file_uploader("Attach documents (PDF or DOCX)", type=["pdf", "docx"], accept_multiple_files=True)
 
     # --- Load models ---
@@ -165,24 +211,29 @@ def main():
 
         # --- Process Uploaded Files ---
         if uploaded_files:
-            for uploaded_file in uploaded_files:
-                file_size = len(uploaded_file.getvalue())
-                if file_size > 1000000000:
-                    st.error(f"{uploaded_file.name}: File size exceeds the 1GB limit.")
-                    continue
-
-                try:
-                    if uploaded_file.type == "application/pdf":
-                        raw_text += extract_text_from_pdf(uploaded_file) + "\n"
-                    elif uploaded_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-                        raw_text += extract_text_from_docx(uploaded_file) + "\n"
-                    else:
-                        st.error(f"{uploaded_file.name}: Unsupported file type")
+            with st.expander("Uploaded Files", expanded=False):
+                for uploaded_file in uploaded_files:
+                    file_size = len(uploaded_file.getvalue())
+                    if file_size > 1000000000:
+                        st.error(f"{uploaded_file.name}: File size exceeds the 1GB limit.")
                         continue
 
-                except Exception as e:
-                    st.error(f"Error processing {uploaded_file.name}: {e}")
-                    continue
+                    try:
+                        if uploaded_file.type == "application/pdf":
+                            extracted_text = extract_text_from_pdf(uploaded_file)
+                            raw_text += extracted_text + "\n"
+                            st.write(f"Extracted text from {uploaded_file.name}")
+                        elif uploaded_file.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+                            extracted_text = extract_text_from_docx(uploaded_file)
+                            raw_text += extracted_text + "\n"
+                            st.write(f"Extracted text from {uploaded_file.name}")
+                        else:
+                            st.error(f"{uploaded_file.name}: Unsupported file type")
+                            continue
+
+                    except Exception as e:
+                        st.error(f"Error processing {uploaded_file.name}: {e}")
+                        continue
 
         # --- Append Manual Text ---
         raw_text += input_text.strip()
@@ -239,4 +290,5 @@ def load_models():
 
 # --- Call Main ---
 if __name__ == "__main__":
+    ai_detection_model, tokenizer, plagiarism_model = load_models()
     main()