Update app.py via AI Editor

app.py

@@ -10,15 +10,17 @@ from dash import html, dcc, Input, Output, State, callback_context, MATCH, ALL
 from docx.shared import Pt
 from docx.enum.style import WD_STYLE_TYPE
 from PyPDF2 import PdfReader
-import openai
 import logging
 import threading
+import requests
+import json
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
 
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP], suppress_callback_exceptions=True)
 
-openai.api_key = os.environ.get("OPENAI_API_KEY", "")
+ANTHROPIC_API_URL = "https://api.anthropic.com/v1/messages"
+ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "YOUR_ANTHROPIC_API_KEY")
 
 uploaded_files = {}
 current_document = None
@@ -242,6 +244,42 @@ def update_right_col(n_clicks_list, btn_ids):
             selected_type = "Shred"
     return get_right_col_content(selected_type)
 
+def anthropic_generate(prompt, model="claude-3-opus-20240229", max_tokens=4096, temperature=0.25):
+    headers = {
+        "x-api-key": ANTHROPIC_API_KEY,
+        "anthropic-version": "2023-06-01",
+        "content-type": "application/json"
+    }
+    data = {
+        "model": model,
+        "max_tokens": max_tokens,
+        "temperature": temperature,
+        "messages": [
+            {"role": "user", "content": prompt},
+        ],
+    }
+    try:
+        response = requests.post(ANTHROPIC_API_URL, headers=headers, data=json.dumps(data), timeout=120)
+        response.raise_for_status()
+        result = response.json()
+        if "content" in result and isinstance(result["content"], list):
+            content = ""
+            for item in result["content"]:
+                if isinstance(item, dict) and "text" in item:
+                    content += item["text"]
+                elif isinstance(item, str):
+                    content += item
+            return content
+        elif "content" in result and isinstance(result["content"], str):
+            return result["content"]
+        elif "stop_reason" in result and "output" in result:
+            return result["output"]
+        else:
+            raise Exception(f"Anthropic API unexpected response: {result}")
+    except Exception as e:
+        logging.error(f"Anthropic API error: {str(e)}")
+        raise
+
 def generate_document(document_type, file_contents):
     prompt = f"""Generate a {document_type} based on the following project artifacts:
 {' '.join(file_contents)}
@@ -273,19 +311,9 @@ like maybe, could be, should, possible be definitive in your language and confid
 to be sure we address them.
 Now, generate the {document_type}:
 """
-
     logging.info(f"Generating document for type: {document_type}")
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4-1106-preview",
-            messages=[
-                {"role": "system", "content": "You are a helpful, expert government proposal writer."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=4096,
-            temperature=0.25,
-        )
-        generated_text = response['choices'][0]['message']['content']
+        generated_text = anthropic_generate(prompt)
         logging.info("Document generated successfully.")
         return generated_text
     except Exception as e:
@@ -419,16 +447,8 @@ Now, provide the updated {document_type}:
 
     logging.info(f"Updating document via chat for {document_type} instruction: {chat_input}")
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4-1106-preview",
-            messages=[
-                {"role": "system", "content": "You are a helpful, expert government proposal writer."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=4096,
-            temperature=0.2,
-        )
-        current_document = response['choices'][0]['message']['content']
+        updated_content = anthropic_generate(prompt, temperature=0.2)
+        current_document = updated_content
         logging.info("Document updated via chat successfully.")
         return f"Document updated based on: {chat_input}", dcc.Markdown(current_document)
     except Exception as e: