Add webcam support

app.py

@@ -1,6 +1,4 @@
 import gradio as gr
-import glob
-import os
 
 from src.core import load_model, predict_traits
 
@@ -12,26 +10,48 @@ TRAIT_NAMES = [
     "Openness",
 ]
 
-DESCRIPTION = "\n'Extraversion': outgoing, energetic, talkative, active, assertive, etc.\n'Neuroticism': worrying, self-pitying, unstable, tense, anxious, etc.\n'Agreeableness': sympathetic, forgiving, generous, kind, appreciative, etc.\n'Conscientiousness': responsible, organized, reliable, efficient, planful, etc.\n'Openness': artistic, curious, imaginative, insightful, original, wide interests, etc."
+DESCRIPTION = [
+    "**Extraversion**: outgoing, energetic, talkative, active, assertive, etc.",
+    "**Neuroticism**: worrying, self-pitying, unstable, tense, anxious, etc.",
+    "**Agreeableness**: sympathetic, forgiving, generous, kind, appreciative, etc.",
+    "**Conscientiousness**: responsible, organized, reliable, efficient, planful, etc.",
+    "**Openness**: artistic, curious, imaginative, insightful, original, wide interests, etc.",
+]
 
 
 def get_traits(video):
     model = load_model()
-    # if webcam_video:
-    #     trait_values = predict_traits(webcam_video, model)
-    # else:
     trait_values = predict_traits(video, model)
     return {k: float(v) for k, v in zip(TRAIT_NAMES, trait_values)}
 
 
-demo = gr.Interface(
+params = dict(
+    # outputs=gr.Label(num_top_classes=5, label="Results"),
+    # title="Personality Traits Prediction 📑",
+    description="Predicts the Big-Five psychology traits of a person based an short introduction video. Adapted from [Deep Impression: Audiovisual Deep Residual Networks for Multimodal Apparent Personality Trait Recognition](https://arxiv.org/abs/1609.05119).",
+    article=" ".join(DESCRIPTION),
+    thumbnail="https://cdn-icons-png.flaticon.com/512/3392/3392044.png",
+    examples="egs",
+    cache_examples=False,
+)
+
+primary_interface = gr.Interface(
     get_traits,
     inputs=gr.Video(label="Video", include_audio=True),
     outputs=gr.Label(num_top_classes=5, label="Results"),
-    title="Personality Traits Prediction [Prototype]",
-    description="Predicts the 5 psychological traits of a person using an short introduction video" + DESCRIPTION,
-    thumbnail="https://cdn-icons-png.flaticon.com/512/3392/3392044.png",
-    examples="egs",
-    cache_examples=True,
+    **params,
+)
+
+second_interface = gr.Interface(
+    get_traits,
+    inputs=gr.Video(label="Webcam", include_audio=True, source="webcam"),
+    outputs=gr.Label(num_top_classes=5, label="Results"),
+    **params,
+)
+
+app = gr.TabbedInterface(
+    [primary_interface, second_interface],
+    title="Personality Traits Prediction 📑",
+    tab_names=["Video Upload", "Webcam"],
 )
-demo.launch()
+app.launch()