Update app.py

app.py

@@ -17,7 +17,7 @@ import plotly.express as px
 import torch
 #import torch_xla.core.xla_model as xm
 from pyannote.audio import Pipeline
-from pyannote.core import Annotation, Segment
+from pyannote.core import Annotation, Segment, Timeline
 from df.enhance import enhance, init_df
 import datetime as dt
 
@@ -226,6 +226,20 @@ for i, tab in enumerate(audio_tabs):
 
             all_dataFrame = su.speakerListToDataFrame(sortedSpeakerList)
             currDF = all_dataFrame
+
+            multiVoice = annotations.get_overlap()
+            singleVoice = annotations.extrude(multiVoice).get_timeline()
+            noVoice = Timeline(segments=[Segment[0,totalSeconds]]).extrude(singleVoice).extrude(multiVoice)
+            # Pie Categories
+            #---------------------------------------------------------------------------
+            df = pd.DataFrame(
+                {
+                    "Duration": [noVoice.duration(),singleVoice.duration(),multiVoice.duration()],
+                    "Category": ["Silence", "Single Voice", "Multiple Voices"],
+                }
+            )
+            fig = px.pie(df, values='Duration', names='Category', title='Types of Discussion')
+            tab.plotly_chart(fig, use_container_width=True)
             
             # Lecturer vs. Audience
             #---------------------------------------------------------------------------