Update app.py

app.py

@@ -30,6 +30,7 @@ from utils import get_translation, welcome, get_large_audio_transcription
 
 from PIL import Image
 
+import stanfordnlp
 
 def main():
  
@@ -41,6 +42,9 @@ def main():
     max = st.sidebar.slider('Select max', 50, 500, step=10, value=150)
     min = st.sidebar.slider('Select min', 10, 450, step=10, value=50)
     gen_summ = False
+    
+ 
+
     with st.spinner("Generating Summary.."):
     
         if button and video:
@@ -60,7 +64,11 @@ def main():
             summ=summarized[0]['summary_text']
             st.write(summ) 
             gen_summ = True
-            
+            stf_nlp = stanfordnlp.Pipeline(processors='tokenize,mwt,pos')
+            doc = stf_nlp(summ)
+            l=[w.text.capitalize() if w.upos in ["PROPN","NNS"] else w.text for sent in doc.sentences for w in sent.words]
+            text=" ".join(l)
+            summ=truecasing_by_sentence_segmentation(text)   
             if 'summary' not in st.session_state:
                 st.session_state.summary=True
                 st.session_state.summarization = summ
@@ -81,8 +89,11 @@ def main():
         translation = get_translation(source='English', dest=s_type, text=summarized_text)
 
         st.sidebar.write(translation)
-    else:
+    elif translate == 'Yes' and gen_summ == False:
         st.error("The summary has not been generated yet. Please generate the summary first and then translate")
+      
+    else:
+         continue 
              
 if __name__ == '__main__':