Update app.py via AI Editor

app.py

@@ -12,16 +12,17 @@ from PyPDF2 import PdfReader
 import logging
 import threading
 import re
-import openai
+import anthropic
 import sqlite3
 import time
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
 
-OPENAI_KEY = os.environ.get("OPENAI_API_KEY", "")
-openai.api_key = OPENAI_KEY
-OPENAI_MODEL = "gpt-3.5-turbo"
-OPENAI_MAX_TOKENS = 32768
+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
 
 DB_PATH = "maiko_dash_docs.sqlite"
 
@@ -646,31 +647,30 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
         else:
             context += f"\n\n{inputs.get('shred','')}"
         full_prompt = f"{prompt}\n\n{context}"
-        logging.info(f"Prompt sent to OpenAI (first 1000 chars): {full_prompt[:1000]}")
+        logging.info(f"Prompt sent to Claude 3 (first 1000 chars): {full_prompt[:1000]}")
         result_holder = {}
 
-        def openai_thread():
+        def anthropic_thread():
             try:
-                response = openai.ChatCompletion.create(
-                    model=OPENAI_MODEL,
+                message = anthropic_client.messages.create(
+                    model=CLAUDE3_SONNET_MODEL,
+                    max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
+                    system="You are a helpful assistant for proposal writing. Always output in markdown, and use markdown tables for spreadsheets if needed.",
                     messages=[
-                        {"role": "system", "content": "You are a helpful assistant for proposal writing. Always output in markdown, and use markdown tables for spreadsheets if needed."},
                         {"role": "user", "content": full_prompt}
-                    ],
-                    max_tokens=OPENAI_MAX_TOKENS,
-                    temperature=0.3,
+                    ]
                 )
-                result_text = response['choices'][0]['message']['content'] if 'choices' in response and response['choices'] and 'message' in response['choices'][0] and 'content' in response['choices'][0]['message'] else str(response)
+                result_text = message.content[0].text if hasattr(message, "content") and message.content else str(message)
                 result_holder['result'] = result_text
-                logging.info("OpenAI document generated successfully.")
+                logging.info("Claude 3 document generated successfully.")
             except Exception as e:
-                logging.error(f"Error with OpenAI API: {str(e)}")
+                logging.error(f"Error with Claude 3 API: {str(e)}")
                 result_holder['result'] = f"Error generating document: {str(e)}"
-        t = threading.Thread(target=openai_thread)
+        t = threading.Thread(target=anthropic_thread)
         t.start()
         t.join(timeout=120)
         generated_doc = result_holder.get('result', 'Error: No document generated.')
-        logging.info(f"OpenAI API returned: {generated_doc[:500]}")  # LOGGING LINE
+        logging.info(f"Claude 3 API returned: {generated_doc[:500]}")
 
         set_generated_doc(selected_type, generated_doc)
         # Preview with fallback