Update app.py

app.py

@@ -10,16 +10,8 @@ qa_pipeline = pipeline("question-answering", model=model_path, tokenizer=model_p
 # Set the title for the Streamlit app
 st.title("Movie Trivia Question Answering")
 
-# Load the context from a predefined file
-context_file_path = 'contexts.txt'  # Replace with the actual path to your context file
-with open(context_file_path, 'r') as file:
-    context = file.read()
-
-# Display the context to the user
-st.subheader("Context (movie-related text)")
-st.write(context)
-
-# Text input for the user to enter the question
+# Text inputs for the user
+context = st.text_area("Enter the context (movie-related text):")
 question = st.text_area("Enter your question:")
 
 def generate_answer(question, context):
@@ -28,15 +20,15 @@ def generate_answer(question, context):
     return result['answer']
 
 if st.button("Get Answer"):
-    if question:
+    if context and question:
         generated_answer = generate_answer(question, context)
         # Display the generated answer
         st.subheader("Answer")
         st.write(generated_answer)
     else:
-        st.warning("Please enter a question.")
+        st.warning("Please enter both context and question.")
 
 # Optionally, add instructions or information about the app
 st.write("""
-    Enter a question related to the provided movie-related context above. The model will provide the answer based on the context.
-    """)
+    Enter a movie-related context and a question related to the context above. The model will provide the answer based on the context provided.
+    """)