Add timestamps

app.py

@@ -11,6 +11,28 @@ from features import FEATURES
 from utils import check_password, process_olivia_data
 
 
+def format_seconds(seconds: int) -> str:
+    if seconds == 1:
+        return "1 second"
+    elif seconds < 60:
+        return f"{seconds} seconds"
+    else:
+        minutes = seconds // 60
+        remaining_seconds = seconds % 60
+
+        if minutes == 1:
+            minute_str = "1 minute"
+        else:
+            minute_str = f"{minutes} minutes"
+
+        if remaining_seconds == 1:
+            second_str = "1 second"
+        else:
+            second_str = f"{remaining_seconds} seconds"
+
+        return f"{minute_str} {second_str}"
+
+
 REPO_URL = "https://huggingface.co/datasets/trevolution/conversation-analytics-comments"
 api = HfApi()
 
@@ -111,7 +133,13 @@ def main():
             st.text('Analytics')
             readable_analytics = ''
             for i, feature in enumerate(analytics):
-                readable_analytics += f"{i+1}. {feature['name']}: {feature['response']}. Quotation: {feature['quotation']}\n\n\n"
+                if feature['timestamp']:
+                    start_time, end_time = int(feature['timestamp'][0]), int(feature['timestamp'][1])
+                    start_time, end_time = format_seconds(start_time), format_seconds(end_time)
+                    
+                    readable_analytics += f"{i+1}. {feature['name']}: {feature['response']}. Quotation: {feature['quotation']}. Timestamp: {start_time}-{end_time}\n\n\n"
+                else:
+                    readable_analytics += f"{i+1}. {feature['name']}: {feature['response']}. Quotation: {feature['quotation']}"
             st.markdown(get_div(readable_analytics), unsafe_allow_html=True)