Update app.py

app.py

@@ -9,6 +9,8 @@ import io
 import requests
 import json
 import openai
+from transformers import AutoConfig, AutoTokenizer, AutoModel
+from summarizer import Summarizer
 
 # File upload size bug?
 
@@ -90,6 +92,10 @@ if st.button('Start Transcription'):
 
         with st.expander("See transcript"):
             st.markdown(transcript)
+# Display the Whisper transcription  
+if 'transcript' in locals():  
+    st.text("Transcription:")  
+    st.text(transcript)  
 
         # Update the user input field with the transcription
         st.session_state.userinput = st.text_area("Input Text:", transcript)
@@ -150,9 +156,10 @@ if st.session_state.userinput and api_key and st.button("Extract Claims", key="c
 
 # Get the extracted claims from Streamlit's session state
 claims_extracted = st.session_state.claims_extraction
-
-from transformers import AutoConfig, AutoTokenizer, AutoModel
-from summarizer import Summarizer
+# Display the Extracted Claims  
+if 'claims_extracted' in st.session_state:  
+    st.text("Extracted Claims:")  
+    st.text(st.session_state.claims_extracted)  
 
 # Define the BERT-based model name
 model_name = 'nlpaueb/legal-bert-base-uncased'
@@ -190,22 +197,11 @@ 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 a text box
-if 'transcript' in locals() and 'claims_extracted' in st.session_state:
-    st.text("Transcription, Extracted Claims, and BERT 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):
-            st.text(f"Summary {i + 1}:\n{summary}")
+# Display the BERT Summaries  
+if summaries:  
+    st.text("BERT Summaries:")  
+    for i, summary in enumerate(summaries):  
+        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)