Update app.py via AI Editor

app.py

@@ -12,18 +12,14 @@ import PyPDF2
 import docx
 import chardet
 
-# Initialize the Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 
-# Get OpenAI API key from Hugging Face Spaces environment variable
 openai.api_key = os.environ.get('OPENAI_API_KEY')
 
-# Global variables
 uploaded_files = {}
 current_matrix = None
 matrix_type = None
 
-# Matrix types and their descriptions
 matrix_types = {
     "Project Deliverables Matrix": "Generate a project deliverables matrix all presumed and actual deliverables based on tasks, requirements and scope.",
     "Communications Plan Matrix": "Create a matrix showing stakeholders, communication methods, frequency, and responsibilities.",
@@ -79,7 +75,7 @@ app.layout = dbc.Container([
             ])
         ], width=3),
         dbc.Col([
-            html.Div(style={"height": "20px"}),  # Added small gap
+            html.Div(style={"height": "20px"}),
             dcc.Loading(
                 id="loading-indicator",
                 type="dot",
@@ -89,7 +85,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"}),  # Added small gap
+            html.Div(style={"height": "20px"}),
             dcc.Loading(
                 id="chat-loading",
                 type="dot",
@@ -124,6 +120,12 @@ def parse_file_content(contents, filename):
         print(f"Error processing file {filename}: {str(e)}")
         return "Error processing file"
 
+def truncate_filename(filename, max_length=24):
+    if len(filename) <= max_length:
+        return filename
+    else:
+        return filename[:max_length - 3] + '...'
+
 @app.callback(
     Output('file-list', 'children'),
     Input('upload-files', 'contents'),
@@ -137,10 +139,47 @@ def update_output(list_of_contents, list_of_names, existing_files):
         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(html.Div([
-                html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
-                html.Span(name)
-            ]))
+            new_files.append(
+                dbc.Card(
+                    dbc.CardBody(
+                        dbc.Row([
+                            dbc.Col(
+                                html.Span(
+                                    truncate_filename(name),
+                                    title=name,
+                                    style={
+                                        'display': 'inline-block',
+                                        'overflow': 'hidden',
+                                        'textOverflow': 'ellipsis',
+                                        'whiteSpace': 'nowrap',
+                                        'maxWidth': '90%',
+                                        'verticalAlign': 'middle',
+                                    }
+                                ),
+                                width='auto',
+                                style={'display': 'flex', 'alignItems': 'center', 'padding': '0'}
+                            ),
+                            dbc.Col(
+                                dbc.Button(
+                                    "Delete",
+                                    id={'type': 'remove-file', 'index': name},
+                                    color="danger",
+                                    size="sm",
+                                    style={'marginLeft': 'auto', 'float': 'right'}
+                                ),
+                                width='auto',
+                                style={'display': 'flex', 'alignItems': 'center', 'justifyContent': 'flex-end', 'padding': '0'}
+                            ),
+                        ],
+                        justify="between",
+                        align="center",
+                        style={"margin": "0", "padding": "0"}
+                        ),
+                        style={'padding': '6px 8px', 'margin': '0', 'display': 'flex', 'alignItems': 'center', 'background': 'none', 'boxShadow': 'none'}
+                    ),
+                    style={'border': 'none', 'boxShadow': 'none', 'background': 'none', 'marginBottom': '2px'}
+                )
+            )
         if existing_files is None:
             existing_files = []
         return existing_files + new_files
@@ -157,9 +196,32 @@ def remove_file(n_clicks, existing_files):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise PreventUpdate
-    removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
+    # Find which button was pressed
+    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
+    # triggered_id is a dict-like string, e.g. "{'type':'remove-file','index':'filename'}"
+    # Safely eval to dict (since dash handles this)
+    import ast
+    try:
+        triggered_id_dict = ast.literal_eval(triggered_id)
+        removed_file = triggered_id_dict['index']
+    except Exception:
+        raise PreventUpdate
     uploaded_files.pop(removed_file, None)
-    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
+    # Filter out the file that was removed
+    filtered_files = []
+    for file_card in existing_files:
+        # file_card is a dict representing a dbc.Card
+        # Traverse to get filename from card body
+        try:
+            # Card > CardBody > Row > [Col, Col] > Col[0] > Span
+            filename_span = file_card['props']['children']['props']['children'][0]['props']['children'][0]['props']['children']
+            # The span text is truncated; check its title for the full name
+            span_title = file_card['props']['children']['props']['children'][0]['props']['children'][0]['props']['title']
+            if span_title != removed_file:
+                filtered_files.append(file_card)
+        except Exception:
+            filtered_files.append(file_card)
+    return filtered_files
 
 def generate_matrix_with_gpt(matrix_type, file_contents):
     prompt = f"""Generate a {matrix_type} based on the following project artifacts:
@@ -188,7 +250,7 @@ Now, generate the {matrix_type}:
     )
 
     matrix_text = response.choices[0].message.content.strip()
-    print("Raw matrix text from GPT:", matrix_text)  # For debugging
+    print("Raw matrix text from GPT:", matrix_text)
 
     lines = [line.strip() for line in matrix_text.split('\n') if '|' in line]
     data = [line.split('|') for line in lines]
@@ -260,7 +322,7 @@ Now, provide the updated {matrix_type}:
     )
 
     updated_matrix_text = response.choices[0].message.content.strip()
-    print("Raw updated matrix text from GPT:", updated_matrix_text)  # For debugging
+    print("Raw updated matrix text from GPT:", updated_matrix_text)
 
     lines = [line.strip() for line in updated_matrix_text.split('\n') if '|' in line]
     data = [line.split('|') for line in lines]
@@ -283,7 +345,6 @@ def download_matrix(n_clicks):
     if current_matrix is None:
         raise PreventUpdate
     
-    # Create an in-memory Excel file
     output = io.BytesIO()
     with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
         current_matrix.to_excel(writer, sheet_name='Sheet1', index=False)