Update app.py via AI Editor

app.py

@@ -199,9 +199,9 @@ def get_left_col_content():
                             'overflowY': 'auto'
                         },
                         rows=5,
-                        wrap='soft'
+                        wrap='soft',
+                        autoFocus=True
                     ),
-                    dcc.Store(id="chat-input-rows", data=5),
                     dbc.Row([
                         dbc.Col(
                             dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3 w-100"),
@@ -292,8 +292,8 @@ def get_right_col_content(selected_type, shred_doc, pink_doc, pink_review_doc, r
                 loaded_preview = markdown_narrative_preview(red_doc) if red_doc else html.Div("No Red document loaded.")
             elif src == "red_review":
                 label = "Red Review"
-                store_var = red_doc
-                loaded_preview = markdown_narrative_preview(red_doc) if red_doc else html.Div("No Red Review document loaded.")
+                store_var = red_review_doc
+                loaded_preview = markdown_table_preview(red_review_doc) if red_review_doc else html.Div("No Red Review document loaded.")
             elif src == "gold":
                 label = "Gold Document"
                 store_var = gold_doc
@@ -359,6 +359,7 @@ app.layout = dbc.Container([
     dcc.Store(id='store-gold-review'),
     dcc.Store(id='store-loe'),
     dcc.Store(id='store-virtual-board'),
+    dcc.Store(id='store-generated-doc'),
     dbc.Row([
         dbc.Col(
             html.H2(id='main-title', className="mt-3 mb-2", style={'textAlign': 'center', 'width':'100%'}),
@@ -477,22 +478,95 @@ def handle_shred_upload(upload_contents, upload_filenames, existing_files, curre
                     )
                 ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
             )
+        logging.info("Shred document uploaded and stored.")
         return files_display, new_shred, markdown_table_preview(new_shred), ""
     return files_display, new_shred, markdown_table_preview(new_shred) if new_shred else html.Div("No Shred document loaded."), ""
 
 @app.callback(
-    Output("chat-input", "rows"),
-    Input("chat-input", "value"),
-    State("chat-input", "rows"),
+    Output('document-preview', 'children'),
+    Output('store-generated-doc', 'data'),
+    Output('loading-output', 'children'),
+    Input({'type': 'btn-generate-doc', 'index': ALL}, 'n_clicks'),
+    State('selected-doc-type', 'data'),
+    State('store-shred', 'data'),
+    State('store-pink', 'data'),
+    State('store-pink-review', 'data'),
+    State('store-red', 'data'),
+    State('store-red-review', 'data'),
+    State('store-gold', 'data'),
+    State('store-gold-review', 'data'),
+    State('store-loe', 'data'),
+    State('store-virtual-board', 'data'),
     prevent_initial_call=True
 )
-def auto_expand_textarea(value, current_rows):
-    if value is None or value == "":
-        return 5
-    num_lines = value.count('\n') + 1
-    max_rows = 20
-    rows = min(max(num_lines, 5), max_rows)
-    return rows
+def generate_document(n_clicks, selected_type, shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board):
+    ctx = callback_context
+    if not ctx.triggered or not any(n_clicks):
+        raise dash.exceptions.PreventUpdate
+    logging.info(f"Generating document for type: {selected_type}")
+    # Gather necessary documents
+    inputs = {
+        "shred": shred,
+        "pink": pink,
+        "pink_review": pink_review,
+        "red": red,
+        "red_review": red_review,
+        "gold": gold,
+        "gold_review": gold_review,
+        "loe": loe,
+        "virtual_board": virtual_board
+    }
+    prompt = document_types.get(selected_type, "")
+    # Compose context for Gemini
+    context = ""
+    if selected_type in doc_dependencies:
+        for dep in doc_dependencies[selected_type]["source"]:
+            context += f"\n\n{dep.upper()}:\n{inputs.get(dep, '')}"
+    else:
+        context += f"\n\n{inputs.get('shred','')}"
+    full_prompt = f"{prompt}\n\n{context}"
+    # Gemini API call (threaded for responsiveness)
+    result_holder = {}
+    def gemini_thread():
+        try:
+            model = genai.GenerativeModel(GEMINI_MODEL)
+            response = model.generate_content(full_prompt, stream=False)
+            result_holder['result'] = response.text if hasattr(response, 'text') else str(response)
+            logging.info("Gemini document generated successfully.")
+        except Exception as e:
+            logging.error(f"Error with Gemini API: {str(e)}")
+            result_holder['result'] = f"Error generating document: {str(e)}"
+    t = threading.Thread(target=gemini_thread)
+    t.start()
+    t.join(timeout=60)
+    generated_doc = result_holder.get('result', 'Error: No document generated.')
+    # Store and preview
+    if selected_type in spreadsheet_types:
+        preview = markdown_table_preview(generated_doc)
+    else:
+        preview = markdown_narrative_preview(generated_doc)
+    logging.info("Document preview updated.")
+    return preview, generated_doc, ""
+
+@app.callback(
+    Output("download-document", "data"),
+    Input("btn-download", "n_clicks"),
+    State('selected-doc-type', 'data'),
+    State('store-generated-doc', 'data'),
+    prevent_initial_call=True
+)
+def download_document(n_clicks, selected_type, generated_doc):
+    if not generated_doc:
+        logging.warning("No generated document to download.")
+        return dash.no_update
+    if selected_type in spreadsheet_types:
+        xlsx_io = markdown_tables_to_xlsx(generated_doc)
+        logging.info("Spreadsheet document prepared for download.")
+        return dcc.send_bytes(xlsx_io.getvalue(), filename=f"{selected_type}_output.xlsx")
+    else:
+        content = generated_doc.encode('utf-8')
+        logging.info("Narrative document prepared for download.")
+        return dcc.send_bytes(content, filename=f"{selected_type}_output.md")
 
 if __name__ == '__main__':
     print("Starting the Dash application...")