Update app.py

app.py

@@ -1,12 +1,11 @@
 import streamlit as st
 import torch
 import torchvision.transforms as transforms
-from torchvision import models
+from torchvision.models.mobilenetv2 import MobileNetV2
 from PIL import Image
 import numpy as np
 import cv2
 from datetime import datetime
-from torchvision.models.mobilenetv2 import MobileNetV2
 from gtts import gTTS
 import os
 
@@ -31,7 +30,7 @@ def speak_streamlit(text):
     tts.save("temp.mp3")
     st.audio("temp.mp3", format="audio/mp3")
 
-# Prediction function
+# Prediction
 def predict(image):
     img = transform(image).unsqueeze(0)
     with torch.no_grad():
@@ -44,44 +43,93 @@ def save_history(image, result):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     image.save(f"history/{timestamp}_{result}.png")
 
-# ====== UI ======
-st.set_page_config(page_title="Currency Detector", page_icon="💵")
-st.title("💵 Currency Authenticity Detector")
-st.subheader("Check if your currency is real or fake!")
-
-currency_type = st.selectbox("Select currency type:", ["PKR (Pakistani Rupees)", "USD (US Dollars)", "INR (Indian Rupees)"])
+# ====== PAGE CONFIG & CUSTOM STYLING ======
+st.set_page_config(page_title="Currency Authenticity Detector", page_icon="💵", layout="centered")
+
+# Custom CSS
+st.markdown("""
+    <style>
+        .main {
+            background-color: #f0f2f6;
+        }
+        .stButton button {
+            background-color: #2e86de;
+            color: white;
+            border-radius: 8px;
+            padding: 0.5em 1.5em;
+            margin: 0.3em 0;
+            transition: all 0.3s ease-in-out;
+        }
+        .stButton button:hover {
+            background-color: #1b4f72;
+            color: white;
+            transform: scale(1.05);
+        }
+        .result-box {
+            font-size: 20px;
+            padding: 10px;
+            border-radius: 10px;
+            background-color: #eaf2f8;
+            text-align: center;
+        }
+        .prediction-fake {
+            color: red;
+            font-weight: bold;
+        }
+        .prediction-real {
+            color: green;
+            font-weight: bold;
+        }
+        .prediction-unknown {
+            color: orange;
+            font-weight: bold;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
+# ====== UI Layout ======
+st.markdown("# 💵 Currency Authenticity Detector")
+st.markdown("### ✅ Instantly check if your currency is **Real or Fake**!")
+
+currency_type = st.selectbox("🔄 Select currency type:", ["PKR (Pakistani Rupees)", "USD (US Dollars)", "INR (Indian Rupees)"])
 if "PKR" not in currency_type:
-    st.warning("Currently only Pakistani Rupees (PKR) is supported. Other currencies coming soon!")
+    st.warning("⚠️ Currently only Pakistani Rupees (PKR) is supported. Other currencies coming soon!")
     st.stop()
 
+st.markdown("### 🖼️ Choose how to scan your currency note:")
+
 # Choose input method
-option = st.radio("Choose method:", ["Upload Image", "Scan via Camera"])
+option = st.radio("🔍 Input Method:", ["Upload Image", "Scan via Camera"])
 
-# ===== Upload Mode =====
+# ===== Upload Image =====
 if option == "Upload Image":
-    uploaded_file = st.file_uploader("Upload a currency image", type=["jpg", "jpeg", "png"])
+    uploaded_file = st.file_uploader("📤 Upload a currency image", type=["jpg", "jpeg", "png"])
     if uploaded_file:
         image = Image.open(uploaded_file).convert("RGB")
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption="🖼️ Uploaded Image", use_column_width=True)
         prediction = predict(image)
 
         if prediction == "Not Currency":
-            st.warning("⚠️ This image does not appear to be a currency note.")
+            st.markdown('<div class="result-box prediction-unknown">⚠️ This does not appear to be a currency note.</div>', unsafe_allow_html=True)
             speak_streamlit("This is not a currency note.")
+        elif prediction == "Fake":
+            st.markdown(f'<div class="result-box prediction-fake">❌ Prediction: {prediction} Currency</div>', unsafe_allow_html=True)
+            speak_streamlit("This is a fake currency note.")
+            save_history(image, prediction)
         else:
-            st.success(f"Prediction: **{prediction}**")
-            speak_streamlit(f"This is a {prediction} currency note.")
+            st.markdown(f'<div class="result-box prediction-real">✔️ Prediction: {prediction} Currency</div>', unsafe_allow_html=True)
+            speak_streamlit("This is a real currency note.")
             save_history(image, prediction)
 
-# ===== Camera Mode =====
+# ===== Webcam Scan =====
 elif option == "Scan via Camera":
-    st.write("Hold the currency in front of your webcam and press Start.")
-    if st.button("Start Camera"):
+    st.markdown("🎥 Press the button below to start your webcam.")
+    start_camera = st.button("📷 Start Camera")
+    if start_camera:
         cap = cv2.VideoCapture(0)
         stframe = st.empty()
         result_box = st.empty()
-
-        stop_button = st.button("Stop Camera")
+        stop_button = st.button("🛑 Stop Camera")
 
         while cap.isOpened():
             ret, frame = cap.read()
@@ -92,13 +140,18 @@ elif option == "Scan via Camera":
             stframe.image(pil_image, channels="RGB", use_column_width=True)
 
             prediction = predict(pil_image)
-            result_box.markdown(f"### Prediction: **{prediction}**")
 
-            if prediction != "Not Currency":
-                speak_streamlit(f"This is a {prediction} currency note.")
+            if prediction == "Not Currency":
+                result_box.markdown('<div class="result-box prediction-unknown">⚠️ This does not appear to be a currency note.</div>', unsafe_allow_html=True)
+                speak_streamlit("This is not a currency note.")
+            elif prediction == "Fake":
+                result_box.markdown(f'<div class="result-box prediction-fake">❌ Prediction: {prediction} Currency</div>', unsafe_allow_html=True)
+                speak_streamlit("This is a fake currency note.")
                 save_history(pil_image, prediction)
             else:
-                speak_streamlit("This is not a currency note.")
+                result_box.markdown(f'<div class="result-box prediction-real">✔️ Prediction: {prediction} Currency</div>', unsafe_allow_html=True)
+                speak_streamlit("This is a real currency note.")
+                save_history(pil_image, prediction)
 
             if stop_button:
                 break
@@ -107,11 +160,16 @@ elif option == "Scan via Camera":
         stframe.empty()
         result_box.empty()
 
-# ===== History =====
+# ===== Scan History =====
 if st.checkbox("📁 Show Scan History"):
-    st.write("Previously scanned images:")
-    for img_file in sorted(os.listdir("history"))[::-1][:5]:
-        st.image(f"history/{img_file}", caption=img_file, width=200)
-
+    st.markdown("### 🕘 Recent Scans")
+    history_files = sorted(os.listdir("history"))[::-1][:5]
+    if not history_files:
+        st.info("No scan history available.")
+    else:
+        for img_file in history_files:
+            st.image(f"history/{img_file}", caption=img_file, width=250)
+
+# ===== Footer =====
 st.markdown("---")
-st.markdown("👨‍💻 Designed by MERAJ GRAPHICS")
+st.markdown("<center>👨‍💻 Designed by <b>MERAJ GRAPHICS</b></center>", unsafe_allow_html=True)