Update pages/Report_Writer.py

pages/Report_Writer.py

@@ -22,7 +22,7 @@ llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.1, model_name="m
 
 # Load and index the input data
 def load_data(document_text):
-    document = [Document(text=document_text)]
+    document = [Document(text=doc_text) for doc_text in document_text]
     
     #index = VectorStoreIndex.from_documents([document])
     index = DocumentSummaryIndex.from_documents(document)
@@ -141,7 +141,7 @@ def main():
     
                 # Load data and generate report
                 doc_list = document_text.split(".")
-                index = load_data(document_text)
+                index = load_data(doc_list)
                 report = generate_report(index, report_format, additional_info)
     
                 st.write("## Generated Report")