Update app.py

app.py

@@ -72,7 +72,7 @@ if st.button('Start Transcription'):
         st.info("Transcribing...")
         st.success("Transcription complete")
         result = model.transcribe(audio_file_path)
-        st.session_state.transcript = result['text']
+        transcript = result['text']  # Define the 'transcript' variable
 
         with st.expander("See transcript"):
             st.markdown(transcript)
@@ -176,21 +176,22 @@ for i, summary in enumerate(summaries):
     st.write(f"### Summary {i+1}")
     st.write(summary)
 
-# Display the Whisper transcription, extracted claims, and BERT summaries in the same text box
-if transcript:
+# Display the Whisper transcription, extracted claims, and BERT summaries in a text box
+if 'transcript' in locals() and 'claims_extracted' in st.session_state:
     st.text("Transcription, Extracted Claims, and BERT Summaries:")
-    
-    # Concatenate the Whisper transcription, extracted claims, and BERT summaries into a single string
-    #results_text = f"Whisper Transcription:\n{transcript}\n\n"
-    #results_text += f"Extracted Claims:\n{claims_extracted}\n\n"
-    results_text += "BERT Summaries:\n"
-    
-    with st.spinner('Generating Summaries...'):
+
+    if 'transcript' in locals():
+        st.text("Transcription:")
+        st.text(transcript)
+
+    if 'claims_extracted' in st.session_state:
+        st.text("Extracted Claims:")
+        st.text(st.session_state.claims_extracted)
+
+    if summaries:
+        st.text("BERT Summaries:")
         for i, summary in enumerate(summaries):
-            results_text += f"Summary {i+1}:\n{summary}\n\n"
-    
-    # Display the concatenated results
-    st.text(results_text)
+            st.text(f"Summary {i + 1}:\n{summary}")
 
 # Citation for the GitHub repo
 st.markdown("<sub>This app was created by [Tonic](https://huggingface.co/tonic) with help from [MIND INTERFACES](https://huggingface.co/MIND-INTERFACES) & (Taylor)[https://huggingface.co/Cloudfaith] [join us on discord](https://discord.gg/5RmtZVVfgQ) </sub>", unsafe_allow_html=True)