Update app.py

app.py

@@ -45,19 +45,12 @@ def process_document(contents: str, filename: str) -> str:
     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:
-        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:
-        document_text = document
-
+def generate_loe(document: str, is_file: bool = False) -> Tuple[str, pd.DataFrame]:
     prompt = f"""
     Analyze the following document and provide a Level of Effort (LOE) breakdown:
 
     Document:
-    {document_text}
+    {document}
 
     For each section header in the document:
     1. Identify the tasks to be completed
@@ -93,6 +86,19 @@ def generate_loe(document: str, is_file: bool = False, filename: str = "") -> Tu
         response_text += "\n\nNote: No detailed LOE table could be generated from the AI response."
     
     return response_text, df
+    
+    # Convert the table to a pandas DataFrame
+    try:
+        if not table_text.strip():
+            raise pd.errors.EmptyDataError("No table found in the response")
+        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
 
 def generate_outline(text: str, instructions: str) -> str:
     prompt = f"""
@@ -129,6 +135,9 @@ def generate_pink_team_document(outline: str, instructions: str) -> str:
     5. Limit the use of bullet points and write predominantly in paragraph format.
     6. Ensure a logical flow of steps taken by MicroHealth for each requirement.
     7. Where applicable, describe the labor category or labor categories that perform the task as part of the process
+    8. Analyze the given requirement for MicroHealth and describe the proposed solution, incorporating relevant industry best practices. 
+    9. Then, provide a detailed explanation of the implementation process, including a step-by-step workflow and any necessary procedures. 
+    10. Conclude by briefly outlining the benefits and measurable outcomes of this approach, citing relevant research to support your claims.
 
     Generate a comprehensive response that showcases MicroHealth's expertise and approach.
     """
@@ -178,7 +187,10 @@ def generate_red_document(document: str, compliance_report: str, instructions: s
     5. Limit the use of bullet points and write predominantly in paragraph format.
     6. Ensure a logical flow of steps taken by MicroHealth for each requirement.
     7. Where applicable, describe the labor category or labor categories that perform the task as part of the process
-    8. {instructions}
+    8. Analyze the given requirement for MicroHealth and describe the proposed solution, incorporating relevant industry best practices. 
+    9. Then, provide a detailed explanation of the implementation process, including a step-by-step workflow and any necessary procedures. 
+    10. Conclude by briefly outlining the benefits and measurable outcomes of this approach, citing relevant research to support your claims.
+
     """
     response = model.generate_content(prompt)
     return response.text
@@ -690,15 +702,15 @@ def update_loe_output(n_clicks, upload_contents, upload_filename, shred_output):
     try:
         if triggered_id in ['generate-loe', 'upload-loe']:
             if upload_contents:
-                loe_text, loe_df = generate_loe(upload_contents, is_file=True, filename=upload_filename)
+                document = process_document(upload_contents, upload_filename)
+                if document.startswith("Unsupported file format") or document.startswith("Error processing document"):
+                    return document
+                loe_text, loe_df = generate_loe(document)
             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")):
-            return loe_text
-        
         return [
             dcc.Markdown(loe_text),
             dash_table.DataTable(