Update app.py

app.py

@@ -29,18 +29,33 @@ def process_document(contents: str, filename: str) -> str:
     content_type, content_string = contents.split(',')
     decoded = base64.b64decode(content_string)
     
-    if filename.endswith('.pdf'):
-        pdf = PdfReader(io.BytesIO(decoded))
-        text = ""
-        for page in pdf.pages:
-            text += page.extract_text()
-    elif filename.endswith('.docx'):
-        doc = Document(io.BytesIO(decoded))
-        text = "\n".join([para.text for para in doc.paragraphs])
+    try:
+        if filename.lower().endswith('.pdf'):
+            pdf = PdfReader(io.BytesIO(decoded))
+            text = ""
+            for page in pdf.pages:
+                text += page.extract_text()
+        elif filename.lower().endswith('.docx'):
+            doc = Document(io.BytesIO(decoded))
+            text = "\n".join([para.text for para in doc.paragraphs])
+        else:
+            return f"Unsupported file format: {filename}. Please upload a PDF or DOCX file."
+        
+        if not text.strip():
+            return "The document appears to be empty. Please check the file and try again."
+        
+        return text
+    except Exception as e:
+        return f"Error processing document: {str(e)}"
+
+def generate_loe(document: str, is_file: bool = False, filename: str = "") -> Tuple[str, pd.DataFrame]:
+    if is_file:
+        # Process the uploaded document
+        document_text = process_document(document, filename)
+        if document_text.startswith("Unsupported file format") or document_text.startswith("Error processing document"):
+            return document_text, pd.DataFrame()
     else:
-        return "Unsupported file format. Please upload a PDF or DOCX file."
-    
-    return text
+        document_text = document
 
 def generate_outline(text: str, instructions: str) -> str:
     prompt = f"""
@@ -612,18 +627,41 @@ def download_g_review(n_clicks, g_review_output):
     State('loe-output', 'children'),
     prevent_initial_call=True,
 )
-def download_loe(n_clicks, loe_output):
-    if loe_output is None:
-        return dash.no_update
-    if isinstance(loe_output, list) and len(loe_output) > 0:
-        content = loe_output[0]['props']['children'] if isinstance(loe_output[0], dict) else str(loe_output[0])
-    else:
-        content = str(loe_output)
-    doc = create_docx(content)
-    buffer = BytesIO()
-    doc.save(buffer)
-    buffer.seek(0)
-    return dcc.send_bytes(buffer.getvalue(), "loe_report.docx")
+
+@app.callback(
+    Output('loe-output', 'children'),
+    Input('generate-loe', 'n_clicks'),
+    State('upload-loe', 'contents'),
+    State('upload-loe', 'filename'),
+    State('shred-output', 'children')
+)
+def update_loe_output(n_clicks, upload_contents, upload_filename, shred_output):
+    if n_clicks is None:
+        return "Click 'Generate LOE' to begin."
+    
+    try:
+        if upload_contents:
+            loe_text, loe_df = generate_loe(upload_contents, is_file=True, filename=upload_filename)
+        elif shred_output:
+            loe_text, loe_df = generate_loe(shred_output)
+        else:
+            return "Please upload a document or complete the Shred tab first."
+        
+        if isinstance(loe_text, str) and loe_text.startswith(("Unsupported file format", "Error processing document", "The document appears to be empty")):
+            return loe_text
+        
+        return [
+            dcc.Markdown(loe_text),
+            dash_table.DataTable(
+                data=loe_df.to_dict('records'),
+                columns=[{'name': i, 'id': i} for i in loe_df.columns],
+                style_table={'overflowX': 'auto'},
+                style_cell={'textAlign': 'left', 'padding': '5px'},
+                style_header={'backgroundColor': 'rgb(230, 230, 230)', 'fontWeight': 'bold'}
+            )
+        ]
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 if __name__ == '__main__':
     print("Starting the Dash application...")