Update app.py

app.py

@@ -1,81 +1,86 @@
-import gradio as gr
-import os
-import json
-
-with open('results.json', 'r') as f:
-    results_data = json.load(f)
-
-with open('sarcasm_data.json', 'r') as f:
-    sarcasm_data = json.load(f)
-
-def get_dialogues_and_result(id):
-    result = next((item for item in results_data if item['id'] == id), None)
-    if not result:
-        return f"ID {id} not found in results.json"
-    
-    sarcasm_info = sarcasm_data.get(id)
-    if not sarcasm_info:
-        return f"ID {id} not found in sarcasm_data.json"
-    
-    utterance = f"{sarcasm_info['speaker']}: {sarcasm_info['utterance']}"
-    context = "\n".join(
-        f"{speaker}: {line}" for speaker, line in zip(sarcasm_info['context_speakers'], sarcasm_info['context'])
-    )
-    
-    output = {
-        "id": id,
-        "utterance": utterance,
-        "context": context,
-        "sarcasm": result['sarcasm']
-    }
-    
-    return output
-
-def analyze_videos(utterance_filename):
-    utterance_id = os.path.splitext(os.path.basename(utterance_filename))[0]
-    output = get_dialogues_and_result(utterance_id)
-    
-    return (
-        output["context"],
-        output["utterance"],
-        output["sarcasm"]
-    )
-
-def custom_css():
-    return """
-    .gradio-heading {
-        text-align: center;
-        color: #0056b3;
-        font-size: 28px;
-        font-weight: bold;
-        margin-bottom: 20px;
-    }
-    """
-
-css_style = custom_css()
-
-with gr.Blocks(css=css_style) as interface:
-    gr.Markdown("<div class='gradio-heading'>Sarcasm Detection Tool</div>")
-    with gr.Row():
-        with gr.Column(scale=1):
-            context_video_input = gr.Video(label="Upload Context Video")
-            punchline_video_input = gr.Video(label="Upload Punchline Video")
-        with gr.Column(scale=2):
-            context_dialogue_output = gr.Textbox(label="Context Dialogues")
-            punchline_dialogue_output = gr.Textbox(label="Punchline Dialogue")
-            punchline_sarcasm_output = gr.Label(label="Sarcasm")
-
-    submit_button = gr.Button("Submit")
-
-    submit_button.click(
-        fn=analyze_videos,
-        inputs=[punchline_video_input],
-        outputs=[
-            context_dialogue_output,
-            punchline_dialogue_output,
-            punchline_sarcasm_output,
-        ]
-    )
-
-if __name__ == "__main__":
-    interface.launch()
+import gradio as gr
+import os
+import json
+
+with open('results.json', 'r') as f:
+    results_data = json.load(f)
+
+with open('sarcasm_data.json', 'r') as f:
+    sarcasm_data = json.load(f)
+
+def get_dialogues_and_result(id):
+    pred = next((item for item in results_data if item['id'] == id), None)
+    if not pred:
+        return f"ID {id} not found in results.json"
+    
+    sarcasm_info = sarcasm_data.get(id)
+    if not sarcasm_info:
+        return f"ID {id} not found in sarcasm_data.json"
+    
+    utterance = f"{sarcasm_info['speaker']}: {sarcasm_info['utterance']}"
+    context = "\n".join(
+        f"{speaker}: {line}" for speaker, line in zip(sarcasm_info['context_speakers'], sarcasm_info['context'])
+    )
+    
+    output = {
+        "id": id,
+        "utterance": utterance,
+        "context": context,
+        "ground_truth": int(sarcasm_info['sarcasm']),
+        "pred": pred['sarcasm']
+    }
+    
+    return output
+
+def analyze_videos(utterance_filename):
+    utterance_id = os.path.splitext(os.path.basename(utterance_filename))[0]
+    output = get_dialogues_and_result(utterance_id)
+    
+    return (
+        output["context"],
+        output["utterance"],
+        output["ground_truth"],
+        output["pred"]
+    )
+
+def custom_css():
+    return """
+    .gradio-heading {
+        text-align: center;
+        color: #0056b3;
+        font-size: 28px;
+        font-weight: bold;
+        margin-bottom: 20px;
+    }
+    """
+
+css_style = custom_css()
+
+with gr.Blocks(css=css_style) as interface:
+    gr.Markdown("<div class='gradio-heading'>Sarcasm Detection</div>")
+    with gr.Row():
+        with gr.Column(scale=1):
+            context_video_input = gr.Video(label="Upload Context Video")
+            punchline_video_input = gr.Video(label="Upload Punchline Video")
+        with gr.Column(scale=2):
+            context_dialogue_output = gr.Textbox(label="Context Dialogues")
+            punchline_dialogue_output = gr.Textbox(label="Punchline Dialogue")
+            with gr.Column():
+                punchline_sarcasm_true = gr.Label(label="Ground Truth")
+                punchline_sarcasm_pred = gr.Label(label="Sarcasm Detected")
+
+    submit_button = gr.Button("Submit")
+
+    submit_button.click(
+        fn=analyze_videos,
+        inputs=[punchline_video_input],
+        outputs=[
+            context_dialogue_output,
+            punchline_dialogue_output,
+            punchline_sarcasm_true,
+            punchline_sarcasm_pred
+        ]
+    )
+
+if __name__ == "__main__":
+    interface.launch()