Update app.py

app.py

@@ -9,52 +9,43 @@ load_dotenv()
 
 IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
 FLAREAI_API = os.getenv("FLAREAI_API")
-VIDEO_API_URL = os.getenv("FLAREAI_VIDEO")
 
-def detect_faces_or_landmarks(input_data, is_video=False):
+def detect_faces(image):
     try:
-        if is_video:
-            video_url = input_data
-            payload = {"video_url": video_url}
-            
-            response = requests.post(VIDEO_API_URL, json=payload)
-            if response.status_code != 200:
-                return f"Video processing failed with status code {response.status_code}: {response.text}"
+        buffered = BytesIO()
+        image.save(buffered, format="JPEG")
+        buffered.seek(0)
 
-            video_url = response.json().get("video_url", "")
-            return gr.Video.update(value=video_url), "Landmarks detected on video!"
-        
-        else:
-            buffered = BytesIO()
-            input_data.save(buffered, format="JPEG")
-            buffered.seek(0)
+        upload_url = "https://api.imgbb.com/1/upload"
+        files = {'image': buffered}
+        data = {"key": IMGBB_API_KEY}
 
-            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}"
 
-            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}")
 
-            image_url = response.json()["data"]["url"]
-            response = requests.post(FLAREAI_API, json={"image_url": image_url})
-            response.raise_for_status()
+        response = requests.post(FLAREAI_API, 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", "")
+        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."
+        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
+        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}", ""
 
+
+# Professional CSS to style the Gradio UI
 css = """
     .gradio-container {
         font-family: 'Roboto', sans-serif;
@@ -123,13 +114,14 @@ css = """
     }
 """
 
+# Gradio Interface
 iface = gr.Interface(
-    fn=detect_faces_or_landmarks,
-    inputs=[gr.Image(type="pil", label="Upload Image"), gr.Video(label="Upload Video URL")],
+    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 & Video Landmark Detection",
-    description="Upload an image or video URL to detect faces and landmarks using AI.",
-    theme="huggingface",
+    title="FlareAI Face Detection",
+    description="Made with ❤️ by Skycodee | Leverage advanced AI for face detection.",
+    theme="huggingface",  # Using Huggingface theme for consistency
     css=css
 )