Update app.py via AI Editor

app.py

@@ -9,38 +9,31 @@ import anthropic
 from threading import Thread
 import logging
 
-# Logging setup
 logging.basicConfig(
     level=logging.INFO,
     format='[%(asctime)s] %(levelname)s - %(message)s'
 )
 
-# Initialize Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
 server = app.server
 
-# Anthropic API setup
 ANTHROPIC_KEY = os.environ.get("ANTHROPIC_API_KEY", "")
 anthropic_client = anthropic.Anthropic(api_key=ANTHROPIC_KEY)
 CLAUDE3_SONNET_MODEL = "claude-3-7-sonnet-20250219"
 CLAUDE3_MAX_CONTEXT_TOKENS = 200_000
 CLAUDE3_MAX_OUTPUT_TOKENS = 64_000
 
-# Global variables
 uploaded_document = None
 shredded_document = None
 generated_response = None
 
-# Helper functions
 def decode_document(decoded_bytes):
     try:
-        # Try UTF-8 first
         content = decoded_bytes.decode('utf-8')
         logging.info("Document decoded as UTF-8.")
         return content
     except UnicodeDecodeError as e_utf8:
         try:
-            # Try latin-1 fallback
             content = decoded_bytes.decode('latin-1')
             logging.warning("Document decoded as Latin-1 due to utf-8 decode error: %s", e_utf8)
             return content
@@ -74,13 +67,13 @@ def process_document(content, filename, action):
             "Do not write as if responding to the proposal:\n\n" + uploaded_document
         )
         def thread_shred():
+            global shredded_document
             try:
                 response = anthropic_client.messages.create(
                     model=CLAUDE3_SONNET_MODEL,
                     max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
                     messages=[{"role": "user", "content": prompt}]
                 )
-                nonlocal shredded_document
                 shredded_document = response.content[0].text
                 logging.info("Document shredded successfully.")
             except Exception as e:
@@ -102,13 +95,13 @@ def process_document(content, filename, action):
             "Refer to research that validates the approach and cite sources with measurable outcomes:\n\n" + shredded_document
         )
         def thread_generate():
+            global generated_response
             try:
                 response = anthropic_client.messages.create(
                     model=CLAUDE3_SONNET_MODEL,
                     max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
                     messages=[{"role": "user", "content": prompt}]
                 )
-                nonlocal generated_response
                 generated_response = response.content[0].text
                 logging.info("Proposal response generated successfully.")
             except Exception as e:
@@ -122,7 +115,6 @@ def process_document(content, filename, action):
     
     return "Action not implemented yet."
 
-# Layout
 app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
@@ -185,7 +177,6 @@ app.layout = dbc.Container([
     ], style={'marginTop':'20px'})
 ], fluid=True)
 
-# Callback for document upload
 @app.callback(
     Output('output-document-upload', 'children'),
     Input('upload-document', 'contents'),
@@ -199,7 +190,6 @@ def handle_upload(content, filename):
         return html.Div(result, style={"wordWrap": "break-word"})
     return ""
 
-# Unified callback for all actions
 @app.callback(
     Output('output-data-upload', 'children'),
     [