Update app.py

app.py

@@ -5,12 +5,8 @@ import pandas as pd
 from transformers import pipeline
 import json
 
-import os
+from datasets import load_dataset
 
-# Create a new directory
-directory_name = "documents"
-if not os.path.exists(directory_name):
-    os.makedirs(directory_name)
 
 
 
@@ -70,19 +66,22 @@ def main():
         
         
         st.image('banner_image.jpg')
-    
+
+
+        
+  
     
     
     elif choice=="Summarizer":
         st.subheader("Text Summarization")
         st.write(" Enter the Text you want to summarize !")
-        documents = {
-            "Document 1": "This is the text for document 1.",
-            "Document 2": "This is the text for document 2.",
-            "Document 3": "This is the text for document 3."
-            }
-        document_name = st.selectbox("Select a document", list(documents.keys()))
-        raw_text = documents[document_name]
+        # Create a dropdown to select which document to read
+        document_names = [str(document['id']) for document in dataset['documents']]
+        document_to_read = st.selectbox("Select a document to read", document_names)
+        # Read the selected document
+        selected_document = dataset['documents'][int(document_to_read)]
+        document_text = selected_document['text']
+        raw_text = documents[document_text]
         num_words = st.number_input("Enter Number of Words in Summary")
         
         if raw_text!="" and num_words is not None: