Update app.py

app.py

@@ -67,22 +67,52 @@ elif option == 'Text summarization':
                 summary = summarizer(text, max_length=130, min_length=30)
                 st.write(summary)
 
+    elif source == "I want to upload a file":
+        uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
+        button = st.button('Get summary')
+        if button:
+            summarizer = summarization_model()
+            with st.spinner(text="Summarizing text..."):
+                summary = summarizer(text, max_length=130, min_length=30)
+                st.write(summary)
+                
 elif option == 'Text generation':
     st.markdown("<h2 style='text-align: center; color:grey;'>Generate text</h2>", unsafe_allow_html=True)
-    text = st.text_input(label='Enter one line of text and let the NLP model generate the rest for you')
-    button = st.button('Generate text')
-    if button:
-        generator = generation_model()
-        with st.spinner(text="Generating text..."):
-            generated_text = generator(text, max_length=50)
-            st.write(generated_text[0]["generated_text"])
+    source = st.radio("How would you like to start? Choose an option below", ["I want to input some text", "I want to upload a file"])
+    if source == "I want to input some text":
+        text = st.text_input(label='Enter one line of text and let the NLP model generate the rest for you')
+        button = st.button('Generate text')
+        if button:
+            generator = generation_model()
+            with st.spinner(text="Generating text..."):
+                generated_text = generator(text, max_length=50)
+                st.write(generated_text[0]["generated_text"])
 
+    elif source == "I want to upload a file":
+        uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
+        button = st.button('Generate text')
+        if button:
+            generator = generation_model()
+            with st.spinner(text="Generating text..."):
+                generated_text = generator(text, max_length=50)
+                st.write(generated_text[0]["generated_text"])
+                
 elif option == 'Sentiment analysis':
     st.markdown("<h2 style='text-align: center; color:grey;'>Classify review</h2>", unsafe_allow_html=True)
-    text = st.text_input(label='Enter a sentence to get its sentiment analysis')
-    button = st.button('Get sentiment analysis')
-    if button:
-        sentiment_analysis = sentiment_model()
-        with st.spinner(text="Getting sentiment analysis..."):
-            sentiment = sentiment_analysis(text)
-            st.write(sentiment[0]["label"])
+    source = st.radio("How would you like to start? Choose an option below", ["I want to input some text", "I want to upload a file"])
+    if source == "I want to input some text":
+        text = st.text_input(label='Enter a sentence to get its sentiment analysis')
+        button = st.button('Get sentiment analysis')
+        if button:
+            sentiment_analysis = sentiment_model()
+            with st.spinner(text="Getting sentiment analysis..."):
+                sentiment = sentiment_analysis(text)
+                st.write(sentiment[0]["label"])
+    elif source == "I want to upload a file":
+        uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
+        button = st.button('Get sentiment analysis')
+        if button:
+            sentiment_analysis = sentiment_model()
+            with st.spinner(text="Getting sentiment analysis..."):
+                sentiment = sentiment_analysis(text)
+                st.write(sentiment[0]["label"])