Update app.py

app.py

@@ -5,36 +5,28 @@ from PIL import Image
 from io import BytesIO
 from dotenv import load_dotenv
 
-load_dotenv()  
-
+load_dotenv()
 
 IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
 FLAREAI_API = os.getenv("FLAREAI_API")
 
 def detect_faces(image):
-    """Send the image to the API and return the processed image and face data."""
     try:
-    
         buffered = BytesIO()
         image.save(buffered, format="JPEG")
         buffered.seek(0)
 
-        
         upload_url = "https://api.imgbb.com/1/upload"
-
-        
         files = {'image': buffered}
         data = {"key": IMGBB_API_KEY}
 
         response = requests.post(upload_url, data=data, files=files)
-
         if response.status_code != 200:
             return f"Image upload failed with status code {response.status_code}: {response.text}"
 
         image_url = response.json()["data"]["url"]
         print(f"Image uploaded successfully: {image_url}")
 
-        
         response = requests.post(FLAREAI_API, json={"image_url": image_url})
         response.raise_for_status()
 
@@ -43,27 +35,68 @@ def detect_faces(image):
         detected_image_url = result.get("image_url", "")
 
         if face_data:
-            face_details = "\n".join([
-                f"Face {index + 1}: Age: {face['age']}, Emotion: {face['emotion']}, Gender: {face['gender']}, Race: {face['race']}"
-                for index, face in enumerate(face_data)
-            ])
+            face_details = "\n".join([f"Face {index + 1}: Age: {face['age']}, Emotion: {face['emotion']}, Gender: {face['gender']}, Race: {face['race']}" for index, face in enumerate(face_data)])
         else:
             face_details = "No faces detected."
 
-        
         result_image = Image.open(BytesIO(requests.get(detected_image_url).content))
-
         return result_image, face_details
     except requests.exceptions.RequestException as e:
         return f"An error occurred: {e}", ""
 
 
+css = """
+    .gradio-container {
+        font-family: 'Arial', sans-serif;
+        background-color: #f9f9f9;
+    }
+    .gradio-title {
+        color: #4e5b66;
+        font-size: 30px;
+        font-weight: 600;
+        text-align: center;
+        margin-top: 30px;
+    }
+    .gradio-description {
+        color: #7a7a7a;
+        font-size: 16px;
+        text-align: center;
+        margin-bottom: 30px;
+    }
+    .gradio-input {
+        border-radius: 12px;
+        box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);
+    }
+    .gradio-output {
+        border-radius: 12px;
+        box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);
+    }
+    .gradio-button {
+        background-color: #4CAF50;
+        color: white;
+        font-weight: 600;
+        border-radius: 12px;
+        padding: 10px 20px;
+        margin-top: 20px;
+        border: none;
+    }
+    .gradio-button:hover {
+        background-color: #45a049;
+    }
+    .gradio-image {
+        border-radius: 12px;
+        border: 2px solid #ccc;
+    }
+"""
+
 iface = gr.Interface(
     fn=detect_faces,
-    inputs=gr.Image(type="pil", label="Upload Image"),
+    inputs=gr.Image(type="pil", label="Upload Image", tool="editor"),
     outputs=[gr.Image(type="pil", label="Processed Image"), gr.Textbox(label="Face Details")],
     title="FlareAI Face Detection",
-    description="Made with ❤️ by skycodee",
+    description="Made with ❤️ by Skycodee",
+    theme="huggingface",
+    css=css
 )
 
 iface.launch()