Update app.py

app.py

@@ -129,7 +129,7 @@ def generate_red_document(document: str, compliance_report: str) -> str:
     response = model.generate_content(prompt)
     return response.text
 
-def generate_loe(document: str) -> Tuple[str, pd.DataFrame]:
+ddef generate_loe(document: str) -> Tuple[str, pd.DataFrame]:
     prompt = f"""
     Analyze the following document and provide a Level of Effort (LOE) breakdown:
 
@@ -148,6 +148,7 @@ def generate_loe(document: str) -> Tuple[str, pd.DataFrame]:
     - Total Hours
 
     Present the detailed breakdown first, followed by the summary table.
+    Ensure the table is properly formatted with | as column separators and a header row.
     """
     response = model.generate_content(prompt)
     
@@ -157,8 +158,13 @@ def generate_loe(document: str) -> Tuple[str, pd.DataFrame]:
     table_text = response.text[table_start:table_end]
     
     # Convert the table to a pandas DataFrame
-    df = pd.read_csv(io.StringIO(table_text), sep='|', skipinitialspace=True).dropna(axis=1, how='all')
-    df.columns = df.columns.str.strip()
+    try:
+        df = pd.read_csv(StringIO(table_text), sep='|', skipinitialspace=True).dropna(axis=1, how='all')
+        df.columns = df.columns.str.strip()
+    except pd.errors.EmptyDataError:
+        # If no table is found or it's empty, create a default DataFrame
+        df = pd.DataFrame(columns=['Task Summary', 'Labor Categories', 'Hours per Labor Category', 'Total Hours'])
+        response.text += "\n\nNote: No detailed LOE table could be generated from the AI response."
     
     return response.text, df