Update app.py

app.py

@@ -129,12 +129,18 @@ 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]:
+def generate_loe(document: str, is_file: bool = False) -> Tuple[str, pd.DataFrame]:
+    if is_file:
+        # Process the uploaded document
+        document_text = process_document(document, document.split(',')[0])
+    else:
+        document_text = document
+
     prompt = f"""
     Analyze the following document and provide a Level of Effort (LOE) breakdown:
 
     Document:
-    {document}
+    {document_text}
 
     For each section header in the document:
     1. Identify the tasks to be completed
@@ -293,6 +299,21 @@ app.layout = dbc.Container([
             dcc.Download(id="download-g-review-doc")
         ]),
         dbc.Tab(label="LOE", tab_id="loe", children=[
+            dcc.Upload(
+                id='upload-loe',
+                children=html.Div(['Drag and Drop or ', html.A('Select Files')]),
+                style={
+                    'width': '100%',
+                    'height': '60px',
+                    'lineHeight': '60px',
+                    'borderWidth': '1px',
+                    'borderStyle': 'dashed',
+                    'borderRadius': '5px',
+                    'textAlign': 'center',
+                    'margin': '10px'
+                },
+                multiple=False
+            ),
             dbc.Button("Generate LOE", id="generate-loe", className="mt-3"),
             dbc.Spinner(html.Div(id='loe-output')),
             dbc.Button("Download LOE Report", id="download-loe", className="mt-3"),
@@ -371,16 +392,19 @@ def update_g_review_output(n_clicks, contents, filename, requirements):
 @app.callback(
     Output('loe-output', 'children'),
     Input('generate-loe', 'n_clicks'),
+    State('upload-loe', 'contents'),
     State('shred-output', 'children')
 )
-def update_loe_output(n_clicks, shred_output):
+def update_loe_output(n_clicks, upload_contents, shred_output):
     if n_clicks is None:
         return "Click 'Generate LOE' to begin."
     
-    if shred_output is None:
-        return "Please complete the Shred tab first."
-    
-    loe_text, loe_df = generate_loe(shred_output)
+    if upload_contents:
+        loe_text, loe_df = generate_loe(upload_contents, is_file=True)
+    elif shred_output:
+        loe_text, loe_df = generate_loe(shred_output)
+    else:
+        return "Please upload a document or complete the Shred tab first."
     
     return [
         dcc.Markdown(loe_text),