Create app.py

app.py

@@ -0,0 +1,69 @@
+import os
+import gradio as gr
+import requests
+from PIL import Image
+from io import BytesIO
+from dotenv import load_dotenv
+
+load_dotenv()  
+
+# Securely fetch your API keys and endpoints
+IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
+CIVILITY_API_URL = 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(CIVILITY_API_URL, json={"image_url": image_url})
+        response.raise_for_status()
+
+        result = response.json()
+        face_data = result.get("face_data", [])
+        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)
+            ])
+        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}", ""
+
+
+iface = gr.Interface(
+    fn=detect_faces,
+    inputs=gr.Image(type="pil", label="Upload Image"),
+    outputs=[gr.Image(type="pil", label="Processed Image"), gr.Textbox(label="Face Details")],
+    title="FlareAI Face Detection",
+    description="Upload an image to detect faces using the CivilityAI API.",
+)
+
+iface.launch()