Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import numpy as np
 from PIL import Image
 from transformers import AutoImageProcessor, AutoModelForImageClassification
+import cv2
+import time
 
 # Load publicly available model from Hugging Face Hub
 processor = AutoImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
@@ -24,14 +26,32 @@ def classify_sign(image):
     prediction = outputs.logits.argmax(-1).item()
     return sign_labels.get(prediction % len(sign_labels), "Unknown Sign")
 
+# Live gesture detection
+def live_detection():
+    cap = cv2.VideoCapture(0)
+    if not cap.isOpened():
+        st.error("Error: Could not open webcam.")
+        return
+    
+    stframe = st.image([])
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+        gesture = classify_sign(image)
+        frame = cv2.putText(frame, f"Detected Gesture: {gesture}", (10, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+        stframe.image(frame, channels="BGR")
+        time.sleep(5)
+    cap.release()
+
 # Streamlit UI
 def main():
     st.set_page_config(page_title="Sign Language Translator", layout="wide")
-    st.title("🤟 Sign Language Translator")
-    st.write("Translate sign language gestures into text and speech.")
+    st.markdown("<h1 style='text-align: center; color: #4CAF50;'>🤟 Sign Language Translator</h1>", unsafe_allow_html=True)
 
     # Top tabs
-    tab1, tab2, tab3 = st.tabs(["Image Load", "Take Picture", "Live"])
+    tab1, tab2, tab3 = st.tabs(["📸 Image Load", "📷 Take Picture", "🎥 Live"])
 
     with tab1:
         uploaded_image = st.file_uploader("Upload an image of a hand gesture", type=["png", "jpg", "jpeg"])
@@ -42,7 +62,7 @@ def main():
                 st.image(image, caption="Uploaded Image", use_column_width=True)
             with col2:
                 gesture = classify_sign(image)
-                st.info(f"Detected Gesture: {gesture}")
+                st.success(f"Detected Gesture: {gesture}")
 
     with tab2:
         camera_image = st.camera_input("Take a picture")
@@ -53,28 +73,26 @@ def main():
                 st.image(image, caption="Captured Image", use_column_width=True)
             with col2:
                 gesture = classify_sign(image)
-                st.info(f"Detected Gesture: {gesture}")
+                st.success(f"Detected Gesture: {gesture}")
 
     with tab3:
-        st.write("**Live gesture detection is now available!**")
+        if st.button("Enable Cam"):
+            live_detection()
 
-    # Left-side tabs
+    # Left-side sidebar with buttons
     with st.sidebar:
-        selected_tab = st.radio("Menu", ["About Us", "Contact Us", "Feedback"])
-
-    if selected_tab == "About Us":
-        st.markdown("**We are team SignAI.** We leverage advanced AI to interpret sign language gestures, making communication more accessible.")
-
-    elif selected_tab == "Contact Us":
-        st.markdown("**Contact Information:**")
-        st.write("📞 Phone: +123-456-7890")
-        st.write("📧 Email: contact@signai.com")
-        st.write("🌐 [LinkedIn](https://linkedin.com)")
-        st.write("📷 [Instagram](https://instagram.com)")
-        st.write("📘 [Facebook](https://facebook.com)")
-
-    elif selected_tab == "Feedback":
-        st.text_area("We value your feedback. Please share your thoughts below:")
+        st.markdown("<h2 style='color: #4CAF50;'>Menu</h2>", unsafe_allow_html=True)
+        if st.button("About Us", key="about_us", help="Learn more about us"):
+            st.markdown("**We are team SignAI.** We leverage advanced AI to interpret sign language gestures, making communication more accessible.")
+        if st.button("Contact Us", key="contact_us", help="Get in touch"):
+            st.markdown("**Contact Information:**")
+            st.write("📞 Phone: +123-456-7890")
+            st.write("📧 Email: contact@signai.com")
+            st.write("🌐 [LinkedIn](https://linkedin.com)")
+            st.write("📷 [Instagram](https://instagram.com)")
+            st.write("📘 [Facebook](https://facebook.com)")
+        if st.button("Feedback", key="feedback", help="Give us your feedback"):
+            st.text_area("We value your feedback. Please share your thoughts below:")
 
 if __name__ == "__main__":
     main()