Update app.py

app.py

@@ -14,7 +14,6 @@ import chardet
 
 # Initialize the Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
-uploaded_files = {}
 
 # Get OpenAI API key from Hugging Face Spaces environment variable
 openai.api_key = os.environ.get('OPENAI_API_KEY')
@@ -26,7 +25,7 @@ matrix_type = None
 
 # Matrix types and their descriptions
 matrix_types = {
-    "Project Mangement Matrix": "Generate a project management matrix summarizing all aspects of project management with components such as scope of work (required), schedule, budget, quality, resources, communications, risk, procurement, and stakeholder management. This is not a project plan but a summary",
+    "Project Mangement Matrix": "Generate a project management matrix outlining all aspects of project management that summarizes components such as scope, schedule, budget, quality, resources, communications, risk, procurement, and stakeholder management.",
     "Communications Plan Matrix": "Create a matrix showing stakeholders, communication methods, frequency, and responsibilities.",
     "Project Kick-off Matrix": "Generate a matrix outlining key project details, goals, team roles, and initial timelines.",
     "Decision Matrix": "Develop a matrix for evaluating options against criteria, with weighted scores analysis of alternatives style.",
@@ -45,10 +44,6 @@ matrix_types = {
     "SWOT Matrix": "Create a matrix analyzing Strengths, Weaknesses, Opportunities, and Threats."
 }
 
-import dash
-from dash import dcc, html
-import dash_bootstrap_components as dbc
-
 app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
@@ -71,10 +66,7 @@ app.layout = dbc.Container([
                 },
                 multiple=True
             ),
-            html.Div([
-                html.H5("Uploaded Files"),
-                dbc.ListGroup(id='file-list')
-            ], className="mt-3"),
+            html.Div(id='file-list'),
             html.Hr(),
             html.Div([
                 dbc.Button(
@@ -87,7 +79,7 @@ app.layout = dbc.Container([
             ])
         ], width=3),
         dbc.Col([
-            html.Div(style={"height": "20px"}),  # Small gap
+            html.Div(style={"height": "20px"}),  # Added small gap
             dcc.Loading(
                 id="loading-indicator",
                 type="dot",
@@ -97,7 +89,7 @@ app.layout = dbc.Container([
             dbc.Button("Download Matrix", id="btn-download", color="success", className="mt-3"),
             dcc.Download(id="download-matrix"),
             html.Hr(),
-            html.Div(style={"height": "20px"}),  # Small gap
+            html.Div(style={"height": "20px"}),  # Added small gap
             dcc.Loading(
                 id="chat-loading",
                 type="dot",
@@ -142,19 +134,18 @@ def update_output(list_of_contents, list_of_names, existing_files):
     global uploaded_files
     if list_of_contents is not None:
         new_files = []
-        for content, name in zip(list_of_contents, list_of_names):
+        for i, (content, name) in enumerate(zip(list_of_contents, list_of_names)):
             file_content = parse_file_content(content, name)
             uploaded_files[name] = file_content
-            new_files.append(dbc.ListGroupItem([
-                dbc.Button("×", id={'type': 'remove-file', 'index': name}, 
-                           color="danger", size="sm", className="me-2"),
+            new_files.append(html.Div([
+                html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
                 html.Span(name)
             ]))
-        existing_files = existing_files or []
+        if existing_files is None:
+            existing_files = []
         return existing_files + new_files
-    return existing_files or []
+    return existing_files
 
-# Update the remove file callback
 @app.callback(
     Output('file-list', 'children', allow_duplicate=True),
     Input({'type': 'remove-file', 'index': dash.ALL}, 'n_clicks'),
@@ -166,23 +157,13 @@ def remove_file(n_clicks, existing_files):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise PreventUpdate
-    
-    triggered_id = ctx.triggered[0]['prop_id']
-    print(f"Triggered ID: {triggered_id}")
-    
-    removed_file = triggered_id.split('"index":')[1].split('"')[1]
-    print(f"Attempting to remove file: {removed_file}")
-    
+    removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
     uploaded_files.pop(removed_file, None)
-    print(f"Remaining files: {uploaded_files}")
-    
-    return [file for file in existing_files if file['props']['children'][1]['children'] != removed_file]
-    
+    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
+
 def generate_matrix_with_gpt(matrix_type, file_contents):
     prompt = f"""Generate a {matrix_type} based on the following project artifacts:
-
 {' '.join(file_contents)}
-
 Instructions:
 1. Create the {matrix_type} as a table.
 2. Use ONLY pipe symbols (|) to separate columns.
@@ -191,19 +172,17 @@ Instructions:
 5. The first row should be the column headers.
 6. Start the output immediately with the column headers.
 7. Each subsequent row should represent a single item in the matrix.
-
 Example format:
 Header1|Header2|Header3
 Item1A|Item1B|Item1C
 Item2A|Item2B|Item2C
-
 Now, generate the {matrix_type}:
 """
 
     response = openai.ChatCompletion.create(
         model="gpt-4-turbo",
         messages=[
-            {"role": "system", "content": "You are a precise matrix generator that outputs only the requested matrix without any additional text. If you get a pws, then use the shall and will statements to extrapolate the information you need, make assumptions."},
+            {"role": "system", "content": "You are a precise matrix generator that outputs only the requested matrix without any additional text."},
             {"role": "user", "content": prompt}
         ]
     )
@@ -227,7 +206,6 @@ Now, generate the {matrix_type}:
     prevent_initial_call=True
 )
 def generate_matrix(*args):
-    print(f"Current uploaded files: {uploaded_files}")
     global current_matrix, matrix_type
     ctx = dash.callback_context
     if not ctx.triggered:
@@ -260,10 +238,8 @@ def update_matrix_via_chat(n_clicks, chat_input):
         raise PreventUpdate
     
     prompt = f"""Update the following {matrix_type} based on this instruction: {chat_input}
-
 Current matrix:
 {current_matrix.to_string(index=False)}
-
 Instructions:
 1. Provide ONLY the updated matrix as a table.
 2. Use ONLY pipe symbols (|) to separate columns.
@@ -272,7 +248,6 @@ Instructions:
 5. The first row should be the column headers.
 6. Start the output immediately with the column headers.
 7. Each subsequent row should represent a single item in the matrix.
-
 Now, provide the updated {matrix_type}:
 """