Update app.py

app.py

@@ -1,8 +1,11 @@
 import streamlit as st
 from helper import *
+from deepface import DeepFace
+import tf_keras
+
 
 # Set the page title
-st.title("Aging Deaging - Assignment 4")
+st.title("Aging Deaging App")
 
 # Create columns for input and output sections
 col1, col2 = st.columns(2)
@@ -26,13 +29,18 @@ with col2:
     st.write(f"Selected conversion: {age_conversion_option}")
     if st.button("Generate"):
         if uploaded_image is not None:
-            # Here you can add your image processing code
-            # For now, we'll just display the uploaded image as a placeholder
-            if age_conversion_option == "Young to Old":
-                processed_image = generate_Y2O(uploaded_image)
-                st.image(processed_image, caption="Old you", use_column_width=True)
-            elif age_conversion_option == "Old to Young":
-                processed_image = generate_O2Y(uploaded_image)
-                st.image(processed_image, caption="Young you", use_column_width=True)
+            image = Image.open(uploaded_image).convert("RGB")  # Convert to RGB
+            image = np.array(image)  # Convert to numpy array for DeepFace and OpenCV
+
+            detections = extract_faces_opencv(image)
+            for face in detections:
+                face_crop = cv2.resize(face, (256, 256))
+                # face_crop = cv2.cvtColor(face_crop, cv2.COLOR_BGR2RGB)
+                if age_conversion_option == "Young to Old":
+                    processed_image = generate_Y2O(face_crop)
+                    st.image(processed_image, caption="Old you", use_container_width=True)
+                elif age_conversion_option == "Old to Young":
+                    processed_image = generate_O2Y(face_crop)
+                    st.image(processed_image, caption="Young you", use_container_width=True)
         else:
             st.warning("Please upload an image before clicking Generate")