Update app.py

app.py

@@ -8,6 +8,9 @@ st.title("Legal Document Analysis")
 st.sidebar.header("Upload Document")
 uploaded_file = st.sidebar.file_uploader("Choose a document", type=["txt"])
 
+# Add text input box for user-provided text
+user_input = st.text_area("Or enter text directly for analysis:")
+
 # Sidebar for selecting task
 st.sidebar.header("Select Task")
 task = st.sidebar.selectbox("Choose the task you want to perform:", ("Summarization", "Named Entity Recognition (NER)"))
@@ -38,15 +41,19 @@ def classify_text(text):
     return result['labels'][0] == "law-related"
 
 # Main area - Display content and perform tasks
-if uploaded_file is not None:
+if uploaded_file is not None or user_input:
     try:
-        # Detect and decode the file content
-        encoding = detect_encoding(uploaded_file)
-        if encoding is None:
-            encoding = 'utf-8'  # Fallback to default encoding
-
-        uploaded_file.seek(0)  # Reset file pointer to the beginning
-        text = uploaded_file.read().decode(encoding)
+        # If a file is uploaded, read the file content
+        if uploaded_file:
+            encoding = detect_encoding(uploaded_file)
+            if encoding is None:
+                encoding = 'utf-8'  # Fallback to default encoding
+
+            uploaded_file.seek(0)  # Reset file pointer to the beginning
+            text = uploaded_file.read().decode(encoding)
+        else:
+            # If no file is uploaded, use user input
+            text = user_input
 
         # Classify the text before proceeding with summarization or NER
         if classify_text(text):
@@ -77,7 +84,7 @@ if uploaded_file is not None:
                     for entity in entities:
                         st.write(f"{entity['entity_group']} - {entity['word']} (Score: {entity['score']:.2f})")
         else:
-            st.warning("The uploaded document does not contain law-related content. Please upload a legal document.")
+            st.warning("The uploaded document or entered text does not contain law-related content. Please provide relevant content.")
 
     except IndexError as e:
         st.error(f"IndexError: {e}. Ensure the text is long enough and parameters are set correctly.")
@@ -88,4 +95,4 @@ if uploaded_file is not None:
     except Exception as e:
         st.error(f"An unexpected error occurred: {e}")
 else:
-    st.info("Please upload a document to analyze.")
+    st.info("Please upload a document or enter text to analyze.")