Update app.py

app.py

@@ -7,22 +7,29 @@ import json
 import uuid
 import base64
 
-# Initialize the Claude client
-client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
-
-# Fallback for older versions of the Anthropic library
-if not hasattr(client, "messages"):
-    # For older versions of the anthropic package
+# Initialize the Claude client - handle different package versions
+try:
+    # Get the version
+    import pkg_resources
+    anthropic_version = pkg_resources.get_distribution("anthropic").version
+    st.sidebar.write(f"Detected Anthropic version: {anthropic_version}")
+    
+    # Initialize based on version
+    if anthropic_version.startswith(("0.3", "0.4", "0.5")):
+        # Older API
+        client = anthropic.Client(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+    else:
+        # Newer API
+        client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+except Exception as e:
+    st.sidebar.write(f"Error detecting Anthropic version: {e}")
+    # Default to older API which is more likely to be compatible
     client = anthropic.Client(api_key=os.environ.get("ANTHROPIC_API_KEY"))
 
 # Set page config
 st.set_page_config(page_title="AI PowerPoint Creator", layout="wide")
 
-# Print Anthropic version for debugging
-try:
-    st.sidebar.write(f"Anthropic version: {anthropic.__version__}")
-except:
-    st.sidebar.write("Couldn't determine Anthropic version")
+# Anthropic version is now checked during client initialization
 
 # Initialize session state variables
 if "current_stage" not in st.session_state:
@@ -72,8 +79,9 @@ def generate_storyboard(title, purpose, audience):
     Start with an engaging title slide and end with a clear call-to-action or summary slide.
     """
     
-    try:
-        # Try newer Anthropic API format first
+    # Check which client type we have
+    if hasattr(client, "messages"):
+        # Newer API
         response = client.messages.create(
             model="claude-3-5-sonnet-20240620",
             max_tokens=4000,
@@ -84,8 +92,8 @@ def generate_storyboard(title, purpose, audience):
             ]
         )
         response_text = response.content[0].text
-    except (AttributeError, TypeError):
-        # Fallback for older Anthropic API
+    else:
+        # Older API
         response = client.completion(
             prompt=f"{anthropic.HUMAN_PROMPT} {prompt} {anthropic.AI_PROMPT}",
             model="claude-2.0",
@@ -130,8 +138,9 @@ def generate_slide_content(slide_info, template):
     Focus on clarity, impact, and alignment with the slide's purpose.
     """
     
-    try:
-        # Try newer Anthropic API format first
+    # Check which client type we have
+    if hasattr(client, "messages"):
+        # Newer API
         response = client.messages.create(
             model="claude-3-5-sonnet-20240620",
             max_tokens=2000,
@@ -142,8 +151,8 @@ def generate_slide_content(slide_info, template):
             ]
         )
         response_text = response.content[0].text
-    except (AttributeError, TypeError):
-        # Fallback for older Anthropic API
+    else:
+        # Older API
         response = client.completion(
             prompt=f"{anthropic.HUMAN_PROMPT} {prompt} {anthropic.AI_PROMPT}",
             model="claude-2.0",
@@ -250,7 +259,7 @@ def render_ideation_stage():
                 if storyboard:
                     st.session_state.storyboard = storyboard
                     st.session_state.current_stage = "storyboard"
-                    st.experimental_rerun()
+                    st.rerun()
 
 def render_storyboard_stage():
     st.header("Step 2: Review and Edit Storyboard")
@@ -296,11 +305,11 @@ def render_storyboard_stage():
     with col1:
         if st.button("⬅️ Back to Ideation"):
             st.session_state.current_stage = "ideation"
-            st.experimental_rerun()
+            st.rerun()
     with col2:
         if st.button("Continue to Template Selection ➡️"):
             st.session_state.current_stage = "template"
-            st.experimental_rerun()
+            st.rerun()
 
 def render_template_stage():
     st.header("Step 3: Select a Template")
@@ -328,7 +337,7 @@ def render_template_stage():
     with col1:
         if st.button("⬅️ Back to Storyboard"):
             st.session_state.current_stage = "storyboard"
-            st.experimental_rerun()
+            st.rerun()
     with col2:
         if st.button("Generate Slides ➡️"):
             st.session_state.current_stage = "slides"
@@ -341,7 +350,7 @@ def render_template_stage():
                     slides_content.append(slide_content)
                 
                 st.session_state.slides_content = slides_content
-                st.experimental_rerun()
+                st.rerun()
 
 def render_slides_stage():
     st.header("Step 4: Review and Edit Slides")
@@ -350,7 +359,7 @@ def render_slides_stage():
         st.warning("No slides content available. Please go back and generate slides.")
         if st.button("⬅️ Back to Template Selection"):
             st.session_state.current_stage = "template"
-            st.experimental_rerun()
+            st.rerun()
         return
     
     # Display slides for review with edit options
@@ -391,11 +400,11 @@ def render_slides_stage():
     with col1:
         if st.button("⬅️ Back to Template Selection"):
             st.session_state.current_stage = "template"
-            st.experimental_rerun()
+            st.rerun()
     with col2:
         if st.button("Create PowerPoint ➡️"):
             st.session_state.current_stage = "export"
-            st.experimental_rerun()
+            st.rerun()
 
 def render_export_stage():
     st.header("Step 5: Export Presentation")