Update app.py

app.py

@@ -20,6 +20,7 @@ import torch
 #import torch_xla.core.xla_model as xm
 from pyannote.audio import Pipeline
 from pyannote.core import Annotation, Segment, Timeline
+from df.enhance import enhance, init_df
 import datetime as dt
 
 enableDenoise = False
@@ -38,22 +39,7 @@ def printV(message,verbosityLevel):
     global verbosity
     if verbosity>=verbosityLevel:
         print(message)
-        
-def get_display_name(speaker, fileIndex):
-    """Return the user-assigned display name for a speaker, or the original label."""
-    renames = st.session_state.speakerRenames
-    if fileIndex < len(renames) and speaker in renames[fileIndex]:
-        return renames[fileIndex][speaker]
-    return speaker
 
-def apply_speaker_renames_to_df(df, fileIndex, column="task"):
-    """Replace speaker_## labels in a DataFrame column with display names."""
-    if column not in df.columns:
-        return df
-    df = df.copy()
-    df[column] = df[column].apply(lambda s: get_display_name(s, fileIndex))
-    return df
-    
 @st.cache_data
 def convert_df(df):
    return df.to_csv(index=False).encode('utf-8')
@@ -158,7 +144,7 @@ def updateCategoryOptions(resultIndex):
     #st.info(f"After update: {st.session_state.categorySelect}")
 
 def updateMultiSelect():
-    currFileIndex = st.session_state.file_names.index(st.session_state["select_currFile"])
+    currFileIndex = file_names.index(st.session_state["select_currFile"])
     st.session_state.resetResult = True
     for i, category in enumerate(st.session_state['categories']):
         st.session_state[f'multiselect_{category}'] = st.session_state['categorySelect'][currFileIndex][i]
@@ -198,113 +184,92 @@ def analyze(inFileName):
             st.session_state.summaries[currFileIndex]["df3"] = df3
             printV(f'Set df3',4)
             
-            # --- Build df4 ---
+            df4_dict = {}
             nameList = st.session_state.categories
             extraNames = []
             valueList = [0 for i in range(len(nameList))]
             extraValues = []
-
+            
             for sp in speakerNames:
                 foundSp = False
                 for i, categoryName in enumerate(nameList):
                     if sp in categorySelections[i]:
+                        #st.info(categoryName)
                         valueList[i] += su.sumTimes(currAnnotation.subset([sp]))
                         foundSp = True
                         break
-                if not foundSp:
+                if foundSp:
+                    continue
+                else:
                     extraNames.append(sp)
                     extraValues.append(su.sumTimes(currAnnotation.subset([sp])))
-
-            if extraNames:
-                extraPairsSorted = sorted(zip(extraNames, extraValues), key=lambda pair: pair[0])
-                extraNames, extraValues = list(zip(*extraPairsSorted))
-                extraNames = list(extraNames)
-                extraValues = list(extraValues)
-            else:
-                extraNames, extraValues = [], []
-
+            extraPairsSorted = sorted(zip(extraNames, extraValues), key=lambda pair: pair[0])
+            extraNames, extraValues = zip(*extraPairsSorted)
             df4_dict = {
-                "values": valueList + extraValues,
-                "names": nameList + extraNames,
-            }
+                    "values": valueList+list(extraValues),
+                    "names": nameList+list(extraNames),
+                }
             df4 = pd.DataFrame(data=df4_dict)
             df4.name = "df4"
             st.session_state.summaries[currFileIndex]["df4"] = df4
-            printV(f'Set df4', 4)
 
-            # --- Build df5 ---
-            speakerList, timeList = su.sumTimesPerSpeaker(oneVoice)
+            printV(f'Set df4',4)
+            
+            speakerList,timeList = su.sumTimesPerSpeaker(oneVoice)
             multiSpeakerList, multiTimeList = su.sumMultiTimesPerSpeaker(multiVoice)
-
-            speakerList = list(speakerList) if speakerList else []
-            timeList = list(timeList) if timeList else []
-            multiSpeakerList = list(multiSpeakerList) if multiSpeakerList else []
-            multiTimeList = list(multiTimeList) if multiTimeList else []
-
-            summativeMultiSpeaker = sum(multiTimeList) if multiTimeList else 1
-            safeOneVoice = sumOneVoice if sumOneVoice > 0 else 1
-
-            basePercentiles = [
-                sumNoVoice / currTotalTime,
-                sumOneVoice / currTotalTime,
-                sumMultiVoice / currTotalTime,
-            ]
-
-            timeStrings = su.timeToString(timeList) if timeList else []
-            multiTimeStrings = su.timeToString(multiTimeList) if multiTimeList else []
-            if isinstance(timeStrings, str):
-                timeStrings = [timeStrings]
-            if isinstance(multiTimeStrings, str):
-                multiTimeStrings = [multiTimeStrings]
-
-            n_ov = len(speakerList)
-            n_mv = len(multiSpeakerList)
-
-            df5 = pd.DataFrame({
-                "ids": ["NV", "OV", "MV"] + [f"OV_{i}" for i in range(n_ov)] + [f"MV_{i}" for i in range(n_mv)],
-                "labels": ["No Voice", "One Voice", "Multi Voice"] + speakerList + multiSpeakerList,
-                "parents": ["", "", ""] + ["OV"] * n_ov + ["MV"] * n_mv,
-                "parentNames": ["Total", "Total", "Total"] + ["One Voice"] * n_ov + ["Multi Voice"] * n_mv,
-                "values": [sumNoVoice, sumOneVoice, sumMultiVoice] + timeList + multiTimeList,
-                "valueStrings": [
-                    su.timeToString(sumNoVoice),
-                    su.timeToString(sumOneVoice),
-                    su.timeToString(sumMultiVoice),
-                ] + timeStrings + multiTimeStrings,
-                "percentiles": [
-                    basePercentiles[0] * 100,
-                    basePercentiles[1] * 100,
-                    basePercentiles[2] * 100,
-                ] + [(t * 100) / safeOneVoice * basePercentiles[1] for t in timeList]
-                  + [(t * 100) / summativeMultiSpeaker * basePercentiles[2] for t in multiTimeList],
-                "parentPercentiles": [
-                    basePercentiles[0] * 100,
-                    basePercentiles[1] * 100,
-                    basePercentiles[2] * 100,
-                ] + [(t * 100) / safeOneVoice for t in timeList]
-                  + [(t * 100) / summativeMultiSpeaker for t in multiTimeList],
-            })
+            summativeMultiSpeaker = sum(multiTimeList)
+            basePercentiles = [sumNoVoice/currTotalTime,
+                                 sumOneVoice/currTotalTime,
+                                 sumMultiVoice/currTotalTime
+                                ]
+            df5 = pd.DataFrame(
+                {
+                    "ids" : ["NV","OV","MV"]+[f"OV_{i}" for i in range(len(speakerList))]
+                                +[f"MV_{i}" for i in range(len(multiSpeakerList))],
+                    "labels" : ["No Voice","One Voice","Multi Voice"] + speakerList + multiSpeakerList,
+                    "parents" : ["","",""]+["OV" for i in range(len(speakerList))]
+                                +["MV" for i in range(len(multiSpeakerList))],
+                    "parentNames" : ["Total","Total","Total"]+["One Voice" for i in range(len(speakerList))]
+                                +["Multi Voice" for i in range(len(multiSpeakerList))],
+                    "values" : [sumNoVoice,
+                                sumOneVoice,
+                                sumMultiVoice,
+                                ] + timeList + multiTimeList,
+                    "valueStrings" : [su.timeToString(sumNoVoice),
+                                su.timeToString(sumOneVoice),
+                                su.timeToString(sumMultiVoice),
+                                ] + su.timeToString(timeList) + su.timeToString(multiTimeList),
+                    "percentiles" : [basePercentiles[0]*100,
+                                     basePercentiles[1]*100,
+                                     basePercentiles[2]*100] +
+                                    [(t*100) / sumOneVoice * basePercentiles[1] for t in timeList] +
+                                    [(t*100) / summativeMultiSpeaker * basePercentiles[2] for t in multiTimeList],
+                    "parentPercentiles" : [basePercentiles[0]*100,
+                                     basePercentiles[1]*100,
+                                     basePercentiles[2]*100] +
+                                    [(t*100) / sumOneVoice for t in timeList] +
+                                    [(t*100) / summativeMultiSpeaker for t in multiTimeList],
+                    
+                }
+            )
             df5.name = "df5"
             st.session_state.summaries[currFileIndex]["df5"] = df5
-            printV(f'Set df5', 4)
-
-            # --- Build speakers_dataFrame, df2 ---
-            speakers_dataFrame, speakers_times = su.annotationToDataFrame(currAnnotation)
+            printV(f'Set df5',4)
+            
+            speakers_dataFrame,speakers_times = su.annotationToDataFrame(currAnnotation)
             st.session_state.summaries[currFileIndex]["speakers_dataFrame"] = speakers_dataFrame
             st.session_state.summaries[currFileIndex]["speakers_times"] = speakers_times
 
             df2_dict = {
-                "values": [100 * t / currTotalTime for t in df4_dict["values"]],
-                "names": df4_dict["names"],
+                "values":[100*t/currTotalTime for t in df4_dict["values"]],
+                "names":df4_dict["names"]
             }
             df2 = pd.DataFrame(df2_dict)
             st.session_state.summaries[currFileIndex]["df2"] = df2
-            printV(f'Set df2', 4)
-    except Exception as e:
-        import traceback
-        print(f"Error in analyze: {e}")
-        traceback.print_exc()
-        st.error(f"Debug - analyze() failed: {e}")
+            printV(f'Set df2',4)
+    except ValueError as e:
+        print(f"Value Error: {e}")
+        pass
 
 #----------------------------------------------------------------------------------------------------------------------
 
@@ -350,8 +315,6 @@ pipeline.to(device)#torch.device("cuda"))
 # Long-range usage
 if 'results' not in st.session_state:
     st.session_state.results = []
-if 'speakerRenames' not in st.session_state:
-    st.session_state.speakerRenames = []
 if 'summaries' not in st.session_state:
     st.session_state.summaries = []
 if 'categories' not in st.session_state:
@@ -425,14 +388,10 @@ if uploaded_file_paths is not None:
         st.session_state.categorySelect.append(tempCategories)
     while (len(st.session_state.summaries) < len(valid_files)):
         st.session_state.summaries.append([])
-    while (len(st.session_state.speakerRenames) < len(valid_files)):
-        st.session_state.speakerRenames.append({})
         
     st.session_state.file_names = file_names
 
 file_names = st.session_state.file_names
-if not file_names:
-    file_names = []
     
 if len(file_names) == 0:
     st.text("Upload file(s) to enable analysis")
@@ -535,8 +494,6 @@ if st.sidebar.button("Load Demo Example"):
         st.session_state.categorySelect.append(tempCategories)
     while (len(st.session_state.summaries) < len(valid_files)):
         st.session_state.summaries.append([])
-    while (len(st.session_state.speakerRenames) < len(valid_files)):
-        st.session_state.speakerRenames.append({})
         
     with st.spinner(text=f'Loading Demo Sample'):
         # RTTM load as filler
@@ -548,8 +505,6 @@ if st.sidebar.button("Load Demo Example"):
                 totalSeconds = segment.end
         st.session_state.results = [(annotations, totalSeconds)]
         st.session_state.summaries = [{}]
-        while len(st.session_state.speakerRenames) < 1:
-            st.session_state.speakerRenames.append({})
         speakerNames = annotations.labels()
         st.session_state.unusedSpeakers = [speakerNames]
         with st.spinner(text=f'Analyzing Demo Data'):
@@ -568,8 +523,6 @@ if currFile is None and len(st.session_state.results) > 0 and len(st.session_sta
     st.write("Select a file to view from the sidebar")
 try:
     st.session_state.resetResult = False
-    if currFile is None:
-        st.stop()
     currFileIndex = file_names.index(currFile)
     currPlainName = currFile.split('.')[0]
     if len(st.session_state.results) > currFileIndex and len(st.session_state.summaries) > currFileIndex and len(st.session_state.results[currFileIndex]) > 0:
@@ -601,27 +554,6 @@ try:
        
         newCategory = st.sidebar.text_input('Add category', key='categoryInput',on_change=addCategory) 
 
-        st.sidebar.divider()
-        st.sidebar.subheader("Rename Speakers")
-        st.sidebar.caption("Replace SPEAKER_## labels with real names.")
-        current_renames = st.session_state.speakerRenames[currFileIndex]
-        with st.sidebar.form("rename_form"):
-            temp_renames = {}
-            for sp in speakerNames:
-                current_label = current_renames.get(sp, "")
-                temp_renames[sp] = st.text_input(
-                    f"{sp}",
-                    value=current_label,
-                    placeholder="e.g. John",
-                )
-            if st.form_submit_button("Apply Names"):
-                for sp, new_name in temp_renames.items():
-                    if new_name.strip():
-                        st.session_state.speakerRenames[currFileIndex][sp] = new_name.strip()
-                    elif sp in st.session_state.speakerRenames[currFileIndex]:
-                        del st.session_state.speakerRenames[currFileIndex][sp]
-                st.rerun()
-
         catTypeColors = su.colorsCSS(3)
         allColors = su.colorsCSS(len(speakerNames)+len(st.session_state.categories))
         speakerColors = allColors[:len(speakerNames)]
@@ -650,8 +582,7 @@ try:
         st.session_state.summaries[currFileIndex]["df4"] = df4
         
         with dataTab:
-            displayDF = apply_speaker_renames_to_df(currDF, currFileIndex, column="Resource")
-            csv = convert_df(displayDF)
+            csv = convert_df(currDF)
             
             st.download_button(
                "Press to Download analysis data",
@@ -661,7 +592,7 @@ try:
                key='download-csv',
                on_click="ignore",
             )
-            st.dataframe(displayDF)
+            st.dataframe(currDF)
         with pie1:
             printV("In Pie1",4)
             df3 = st.session_state.summaries[currFileIndex]["df3"]
@@ -675,52 +606,46 @@ try:
             printV("Pie1 Pretrace",4)
             fig1.add_trace(go.Pie(values=df3["values"],labels=df3["names"],sort=False))
             printV("Pie1 Posttrace",4)
-            st.plotly_chart(fig1, use_container_width=True, config=config)
+            
             col1_1, col1_2 = st.columns(2)
-            try:
-                fig1.write_image("ascn_pie1.pdf")
-                fig1.write_image("ascn_pie1.svg")
-            except Exception:
-                pass
+            fig1.write_image("ascn_pie1.pdf")
+            fig1.write_image("ascn_pie1.svg")
             printV("Pie1 files written",4)
             with col1_1:
-                if os.path.exists('ascn_pie1.pdf'):
-                    printV("Pie1 in col1_1",4)
-                    with open('ascn_pie1.pdf','rb') as f:
-                        printV("Pie1 in file open",4)
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-voice-category-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf1',
-                            on_click="ignore",
-                        )
-                        printV("Pie1 after col1_1",4)
+                printV("Pie1 in col1_1",4)
+                with open('ascn_pie1.pdf','rb') as f:
+                    printV("Pie1 in file open",4)
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-voice-category-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf1',
+                        on_click="ignore",
+                    )
+                    printV("Pie1 after col1_1",4)
             with col1_2:
-                if os.path.exists('ascn_pie1.svg'):
-                    with open('ascn_pie1.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-voice-category-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg1',
-                            on_click="ignore",
-                        )
-                        printV("Pie1 in col1_2",4)
+                with open('ascn_pie1.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-voice-category-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg1',
+                        on_click="ignore",
+                    )
+                    printV("Pie1 in col1_2",4)
+            st.plotly_chart(fig1, use_container_width=True,config=config)
             printV("Pie1 post plotly",4)
 
         with pie2:
-            printV("In Pie2",4)
-            df4 = st.session_state.summaries[currFileIndex]["df4"].copy()
+            df4 = st.session_state.summaries[currFileIndex]["df4"]
         
             # Some speakers may be missing, so fix colors
             figColors = []
             for n in df4["names"]:
                 if n in speakerNames:
                     figColors.append(speakerColors[speakerNames.index(n)])
-            df4["names"] = df4["names"].apply(lambda s: get_display_name(s, currFileIndex))
             fig2 = go.Figure()
             fig2.update_layout(
                 title_text="Percentage of Speakers and Custom Categories",
@@ -728,43 +653,35 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
-            printV("Pie2 Pretrace",4)
             fig2.add_trace(go.Pie(values=df4["values"],labels=df4["names"],sort=False))
-            printV("Pie2 Posttrace",4)
-            st.plotly_chart(fig2, use_container_width=True, config=config)
+
             col2_1, col2_2 = st.columns(2)
-            try:
-                fig2.write_image("ascn_pie2.pdf")
-                fig2.write_image("ascn_pie2.svg")
-            except Exception:
-                pass
+            fig2.write_image("ascn_pie2.pdf")
+            fig2.write_image("ascn_pie2.svg")
             with col2_1:
-                if os.path.exists('ascn_pie2.pdf'):
-                    with open('ascn_pie2.pdf','rb') as f:
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-speaker-percent-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf2',
-                            on_click="ignore",
-                        )
+                with open('ascn_pie2.pdf','rb') as f:
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-speaker-percent-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf2',
+                        on_click="ignore",
+                    )
             with col2_2:
-                if os.path.exists('ascn_pie2.svg'):
-                    with open('ascn_pie2.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-speaker-percent-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg2',
-                            on_click="ignore",
-                        )
+                with open('ascn_pie2.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-speaker-percent-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg2',
+                        on_click="ignore",
+                    )
+            st.plotly_chart(fig2, use_container_width=True,config=config)
 
         with sunburst1:
-            df5 = st.session_state.summaries[currFileIndex]["df5"].copy()
-            df5["labels"] = df5["labels"].apply(lambda s: get_display_name(s, currFileIndex))
-            df5["parentNames"] = df5["parentNames"].apply(lambda s: get_display_name(s, currFileIndex))
+            df5 = st.session_state.summaries[currFileIndex]["df5"]
             fig3_1 = px.sunburst(df5,
                                 branchvalues = 'total',
                                 names = "labels",
@@ -789,40 +706,34 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
-            st.plotly_chart(fig3_1, use_container_width=True, config=config)
+            
             col3_1, col3_2 = st.columns(2)
-            try:
-                fig3_1.write_image("ascn_sunburst.pdf")
-                fig3_1.write_image("ascn_sunburst.svg")
-            except Exception:
-                pass
+            fig3_1.write_image("ascn_sunburst.pdf")
+            fig3_1.write_image("ascn_sunburst.svg")
             with col3_1:
-                if os.path.exists('ascn_sunburst.pdf'):
-                    with open('ascn_sunburst.pdf','rb') as f:
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-speaker-categories-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf3',
-                            on_click="ignore",
-                        )
+                with open('ascn_sunburst.pdf','rb') as f:
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-speaker-categories-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf3',
+                        on_click="ignore",
+                    )
             with col3_2:
-                if os.path.exists('ascn_sunburst.svg'):
-                    with open('ascn_sunburst.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-speaker-categories-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg3',
-                            on_click="ignore",
-                        )
+                with open('ascn_sunburst.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-speaker-categories-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg3',
+                        on_click="ignore",
+                    )
+            st.plotly_chart(fig3_1, use_container_width=True,config=config)
     
         with treemap1:
-            df5 = st.session_state.summaries[currFileIndex]["df5"].copy()
-            df5["labels"] = df5["labels"].apply(lambda s: get_display_name(s, currFileIndex))
-            df5["parentNames"] = df5["parentNames"].apply(lambda s: get_display_name(s, currFileIndex))
+            df5 = st.session_state.summaries[currFileIndex]["df5"]
             fig3 = px.treemap(df5,
                             branchvalues = "total",
                             names = "labels",
@@ -847,43 +758,37 @@ try:
                 plot_bgcolor='rgba(0, 0, 0, 0)',
                 paper_bgcolor='rgba(0, 0, 0, 0)',
             )
-            st.plotly_chart(fig3, use_container_width=True, config=config)            
+                        
             col4_1, col4_2 = st.columns(2)
-            try:
-                fig3.write_image("ascn_treemap.pdf")
-                fig3.write_image("ascn_treemap.svg")
-            except Exception:
-                pass
+            fig3.write_image("ascn_treemap.pdf")
+            fig3.write_image("ascn_treemap.svg")
             with col4_1:
-                if os.path.exists('ascn_treemap.pdf'):
-                    with open('ascn_treemap.pdf','rb') as f:
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-treemap-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf4',
-                            on_click="ignore",
-                        )
+                with open('ascn_treemap.pdf','rb') as f:
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-treemap-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf4',
+                        on_click="ignore",
+                    )
             with col4_2:
-                if os.path.exists('ascn_treemap.svg'):
-                    with open('ascn_treemap.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-treemap-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg4',
-                            on_click="ignore",
-                        )
+                with open('ascn_treemap.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-treemap-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg4',
+                        on_click="ignore",
+                    )
+            st.plotly_chart(fig3, use_container_width=True,config=config)
        
         # generate plotting window
 
 
         with timeline:
-            timeline_df = speakers_dataFrame.copy()
-            timeline_df["Resource"] = timeline_df["Resource"].apply(lambda s: get_display_name(s, currFileIndex))
-            fig_la = px.timeline(timeline_df, x_start="Start", x_end="Finish", y="Resource", color="Resource",title="Timeline of Audio with Speakers",
+            fig_la = px.timeline(speakers_dataFrame, x_start="Start", x_end="Finish", y="Resource", color="Resource",title="Timeline of Audio with Speakers",
                                 color_discrete_sequence=speakerColors)
             fig_la.update_yaxes(autorange="reversed") 
             
@@ -909,39 +814,34 @@ try:
                 legend={'traceorder':'reversed'},
                 yaxis= {'showticklabels': False},
             )
-            st.plotly_chart(fig_la, use_container_width=True, config=config)                        
+                                    
             col5_1, col5_2 = st.columns(2)
-            try:
-                fig_la.write_image("ascn_timeline.pdf")
-                fig_la.write_image("ascn_timeline.svg")
-            except Exception:
-                pass
+            fig_la.write_image("ascn_timeline.pdf")
+            fig_la.write_image("ascn_timeline.svg")
             with col5_1:
-                if os.path.exists('ascn_timeline.pdf'):
-                    with open('ascn_timeline.pdf','rb') as f:
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-timeline-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf5',
-                            on_click="ignore",
-                        )
+                with open('ascn_timeline.pdf','rb') as f:
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-timeline-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf5',
+                        on_click="ignore",
+                    )
             with col5_2:
-                if os.path.exists('ascn_timeline.svg'):
-                    with open('ascn_timeline.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-timeline-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg5',
-                            on_click="ignore",
-                        )
+                with open('ascn_timeline.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-timeline-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg5',
+                        on_click="ignore",
+                    )
+            st.plotly_chart(fig_la, use_container_width=True,config=config)
     
         with bar1:
-            df2 = st.session_state.summaries[currFileIndex]["df2"].copy()
-            df2["names"] = df2["names"].apply(lambda s: get_display_name(s, currFileIndex))
+            df2 = st.session_state.summaries[currFileIndex]["df2"]
             fig2_la = px.bar(df2, x="values", y="names", color="names", orientation='h',
                             custom_data=["names","values"],title="Time Spoken by each Speaker",
                             color_discrete_sequence=catColors+speakerColors)
@@ -962,40 +862,34 @@ try:
                     'Percentage of Time: %{customdata[1]:.2f}%'
                 ])
             )
-            st.plotly_chart(fig2_la, use_container_width=True, config=config)                                    
+                                                
             col6_1, col6_2 = st.columns(2)
-            try:
-                fig2_la.write_image("ascn_bar.pdf")
-                fig2_la.write_image("ascn_bar.svg")
-            except Exception:
-                pass
+            fig_la.write_image("ascn_bar.pdf")
+            fig_la.write_image("ascn_bar.svg")
             with col6_1:
-                if os.path.exists('ascn_bar.pdf'):
-                    with open('ascn_bar.pdf','rb') as f:
-                        st.download_button(
-                            "Save As PDF",
-                            f,
-                            'sonogram-speaker-time-'+currPlainName+'.pdf',
-                            'application/pdf',
-                            key='download-pdf6',
-                            on_click="ignore",
-                        )
+                with open('ascn_bar.pdf','rb') as f:
+                    st.download_button(
+                        "Save As PDF",
+                        f,
+                        'sonogram-speaker-time-'+currPlainName+'.pdf',
+                        'application/pdf',
+                        key='download-pdf6',
+                        on_click="ignore",
+                    )
             with col6_2:
-                if os.path.exists('ascn_bar.svg'):
-                    with open('ascn_bar.svg','rb') as f:
-                        st.download_button(
-                            "Save As SVG",
-                            f,
-                            'sonogram-speaker-time-'+currPlainName+'.svg',
-                            'image/svg+xml',
-                            key='download-svg6',
-                            on_click="ignore",
-                        )
-
-except ValueError as e:
-    import traceback
-    st.error(f"Rendering error: {e}")
-    traceback.print_exc()
+                with open('ascn_bar.svg','rb') as f:
+                    st.download_button(
+                        "Save As SVG",
+                        f,
+                        'sonogram-speaker-time-'+currPlainName+'.svg',
+                        'image/svg+xml',
+                        key='download-svg6',
+                        on_click="ignore",
+                    )
+            st.plotly_chart(fig2_la, use_container_width=True,config=config)
+
+except ValueError:
+    pass
 
 if len(st.session_state.results) > 0:
     with st.expander("Multi-file Summary Data"):