Update app.py

app.py

@@ -46,16 +46,19 @@ if option == "Extractive question answering":
             with st.spinner(text="Getting answer..."):
                 answer = question_answerer(context=context, question=question)
                 answer = answer["answer"]
-                st.write(f"Answer: {answer}")
+                st.write(f"Answer ---> {answer}")
     elif source == "I want to upload a file":
         uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
-        question = st.text_input(label='Enter your question')
-        button = st.button('Get answer')
+        if uploaded_file is not None:
+            sample_text = uploaded_file.read()
+            context = st.text_area("", value=sample_text, height=330)
+        question = st.text_input(label="Enter your question")
+        button = st.button("Get answer")
         if button:
             question_answerer = question_model()
             with st.spinner(text="Getting answer..."):
-                answer = question_answerer(context=uploaded_file, question=question)
-                st.write(f"Answer: {answer}")
+                answer = question_answerer(context=context, question=question)
+                st.write(f"Answer ---> {answer}")
 
 elif option == 'Text summarization':
     st.markdown("<h2 style='text-align: center; color:red;'>Summarize text</h2>", unsafe_allow_html=True)