Create app.py

app.py

@@ -0,0 +1,65 @@
+import streamlit as st
+from PIL import Image
+import openai
+import cv2
+import numpy as np
+import tensorflow as tf
+
+# Set OpenAI API key
+openai.api_key = "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"
+
+# Title of the app
+st.title("AttraVision - Compare and Analyze Attractiveness")
+
+# Uploading the images
+st.header("Upload Two Photos for Comparison")
+uploaded_file1 = st.file_uploader("Choose the first image", type=["jpg", "jpeg", "png"])
+uploaded_file2 = st.file_uploader("Choose the second image", type=["jpg", "jpeg", "png"])
+
+# Helper function to process image
+def process_image(image_file):
+    img = Image.open(image_file)
+    img = img.resize((224, 224))  # Resize image for consistent input size
+    img_array = np.array(img)
+    return img_array
+
+# Helper function to analyze image using OpenAI
+def analyze_image(image_array):
+    # Convert the image to bytes to send to OpenAI API
+    img_bytes = Image.fromarray(image_array).tobytes()
+    
+    # Use OpenAI's image recognition model to analyze the photo (vision capabilities)
+    response = openai.Image.create(file=img_bytes)
+    # You can extract specific insights such as facial features or symmetry
+    return response["data"]  # Example response handling
+
+# Analyze and compare the two photos
+if uploaded_file1 and uploaded_file2:
+    # Display the uploaded images
+    st.image([uploaded_file1, uploaded_file2], caption=["Photo 1", "Photo 2"], width=300)
+
+    # Process and analyze the images
+    img1 = process_image(uploaded_file1)
+    img2 = process_image(uploaded_file2)
+
+    st.write("Analyzing photos...")
+    
+    # Analyze the first image
+    analysis1 = analyze_image(img1)
+    st.write("Photo 1 Analysis: ", analysis1)
+
+    # Analyze the second image
+    analysis2 = analyze_image(img2)
+    st.write("Photo 2 Analysis: ", analysis2)
+
+    # Example of comparison logic (this could be extended for metrics like symmetry)
+    if analysis1["symmetry_score"] > analysis2["symmetry_score"]:
+        st.write("Photo 1 is more symmetrical than Photo 2.")
+    else:
+        st.write("Photo 2 is more symmetrical than Photo 1.")
+
+    # Provide more detailed metrics such as emotion, lighting, etc.
+    st.write("Additional metrics such as emotion, lighting, etc., will be added here.")
+
+else:
+    st.write("Please upload two images to compare.")