Update app.py via AI Editor

app.py

@@ -496,70 +496,164 @@ def handle_shred_upload(contents, filenames, current_shred, file_list):
     return latest_text, file_previews, filenames[-1]
 
 @app.callback(
-    [
-        Output('store-pink', 'data'),
-        Output('store-pink-review', 'data'),
-        Output('store-red', 'data'),
-        Output('store-red-review', 'data'),
-        Output('store-gold', 'data'),
-        Output('store-gold-review', 'data'),
-        Output('store-loe', 'data'),
-        Output('store-virtual-board', 'data'),
-        Output('uploaded-doc-name-pink', 'children'),
-        Output('uploaded-doc-name-pink_review', 'children'),
-        Output('uploaded-doc-name-red', 'children'),
-        Output('uploaded-doc-name-red_review', 'children'),
-        Output('uploaded-doc-name-gold', 'children'),
-        Output('uploaded-doc-name-gold_review', 'children'),
-        Output('uploaded-doc-name-loe', 'children'),
-        Output('uploaded-doc-name-virtual_board', 'children'),
-    ],
-    [
-        Input({'type': 'upload-doc-type', 'subtype': ALL, 'index': ALL}, 'contents')
-    ],
-    [
-        State({'type': 'upload-doc-type', 'subtype': ALL, 'index': ALL}, 'filename')
-    ],
+    Output('store-pink', 'data'),
+    Output('uploaded-doc-name-pink', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'pink', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'pink', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_pink_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-pink-review', 'data'),
+    Output('uploaded-doc-name-pink_review', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'pink_review', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'pink_review', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_pink_review_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-red', 'data'),
+    Output('uploaded-doc-name-red', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'red', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'red', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_red_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-red-review', 'data'),
+    Output('uploaded-doc-name-red_review', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'red_review', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'red_review', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_red_review_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-gold', 'data'),
+    Output('uploaded-doc-name-gold', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'gold', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'gold', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_gold_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-gold-review', 'data'),
+    Output('uploaded-doc-name-gold_review', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'gold_review', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'gold_review', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_gold_review_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-loe', 'data'),
+    Output('uploaded-doc-name-loe', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'loe', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'loe', 'index': ALL}, 'filename'),
+    prevent_initial_call=True
+)
+def handle_loe_upload(contents_list, filenames_list):
+    if not contents_list or not filenames_list:
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
+        if contents and filenames:
+            if isinstance(contents, str):
+                contents = [contents]
+                filenames = [filenames]
+            for content, name in zip(contents, filenames):
+                file_text = process_document(content, name)
+                return file_text, name
+    return dash.no_update, dash.no_update
+
+@app.callback(
+    Output('store-virtual-board', 'data'),
+    Output('uploaded-doc-name-virtual_board', 'children'),
+    Input({'type': 'upload-doc-type', 'subtype': 'virtual_board', 'index': ALL}, 'contents'),
+    State({'type': 'upload-doc-type', 'subtype': 'virtual_board', 'index': ALL}, 'filename'),
     prevent_initial_call=True
 )
-def handle_dep_upload_to_stores(contents_list, filenames_list):
-    stores = [dash.no_update]*8
-    names = [dash.no_update]*8
+def handle_virtual_board_upload(contents_list, filenames_list):
     if not contents_list or not filenames_list:
-        return stores + names
-    for i, (contents, filenames) in enumerate(zip(contents_list, filenames_list)):
+        return dash.no_update, dash.no_update
+    for contents, filenames in zip(contents_list, filenames_list):
         if contents and filenames:
             if isinstance(contents, str):
                 contents = [contents]
                 filenames = [filenames]
             for content, name in zip(contents, filenames):
                 file_text = process_document(content, name)
-                # Assign to correct store and name output
-                if 'pink_review' in name.lower():
-                    stores[1] = file_text
-                    names[1] = name
-                elif 'pink' in name.lower() and 'review' not in name.lower():
-                    stores[0] = file_text
-                    names[0] = name
-                elif 'red_review' in name.lower():
-                    stores[3] = file_text
-                    names[3] = name
-                elif 'red' in name.lower() and 'review' not in name.lower():
-                    stores[2] = file_text
-                    names[2] = name
-                elif 'gold_review' in name.lower():
-                    stores[5] = file_text
-                    names[5] = name
-                elif 'gold' in name.lower() and 'review' not in name.lower():
-                    stores[4] = file_text
-                    names[4] = name
-                elif 'loe' in name.lower():
-                    stores[6] = file_text
-                    names[6] = name
-                elif 'virtual' in name.lower():
-                    stores[7] = file_text
-                    names[7] = name
-    return stores + names
+                return file_text, name
+    return dash.no_update, dash.no_update
 
 @app.callback(
     Output('document-preview', 'children'),