Update app.py

app.py

@@ -78,41 +78,20 @@ elif option == 'Text summarization':
                 
 elif option == 'Text generation':
     st.markdown("<h2 style='text-align: center; color:grey;'>Generate text</h2>", unsafe_allow_html=True)
-    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"])
+    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 option == 'Sentiment analysis':
     st.markdown("<h2 style='text-align: center; color:grey;'>Classify review</h2>", unsafe_allow_html=True)
-    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"])
+    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"])