Update app.py

app.py

@@ -9,71 +9,77 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_bluesky_post(arg: str) -> str:  # Keeping the original function signature
-    """Fetches the most recent post from a Bluesky account. 
+def get_bluesky_post(arg: str) -> str:
+    """Fetches the most recent post from a Bluesky account.
     Args:
         arg: The Bluesky handle without the domain (e.g., 'beccalewy')
     Returns:
         A string containing the most recent post content or an error message
     """
     import requests
-    import json
     
     # Format handle properly
-    handle = arg  # Using arg as the handle
-    if not "." in handle:
+    handle = arg.strip()  # Using arg as the handle and removing whitespace
+    if "." not in handle:
         full_handle = f"{handle}.bsky.social"
     else:
         full_handle = handle
     
     try:
-        # First, we need to use the public API that doesn't require authentication
-        # This API endpoint lets us get public posts without authentication
-        search_url = f"https://api.bsky.app/search/profiles?q={handle}"
-        response = requests.get(search_url)
+        # Step 1: Use the public API endpoint to resolve the handle to a DID
+        # As per docs: public endpoints can be called directly against https://public.api.bsky.app
+        resolver_url = "https://public.api.bsky.app/xrpc/com.atproto.identity.resolveHandle"
+        resolver_params = {"handle": full_handle}
         
-        if response.status_code != 200:
-            return f"Error: Could not search for the profile. Status code: {response.status_code}"
+        resolver_response = requests.get(resolver_url, params=resolver_params)
         
-        profiles = response.json()
-        if not profiles or "profiles" not in profiles or len(profiles["profiles"]) == 0:
-            return f"No Bluesky account found for @{handle}"
-            
-        # Find the matching profile
-        matching_profile = None
-        for profile in profiles["profiles"]:
-            if profile["handle"].lower() == full_handle.lower():
-                matching_profile = profile
-                break
-                
-        if not matching_profile:
-            return f"Could not find an exact match for @{handle}"
-            
-        # Use the profile's handle to get their feed through the public API
-        feed_url = f"https://api.bsky.app/profile/{matching_profile['handle']}/feed"
-        feed_response = requests.get(feed_url)
+        if resolver_response.status_code != 200:
+            return f"Could not resolve handle @{handle}. Status code: {resolver_response.status_code}"
+        
+        resolver_data = resolver_response.json()
+        user_did = resolver_data.get("did")
+        
+        if not user_did:
+            return f"Could not find DID for Bluesky account @{handle}"
+        
+        # Step 2: Get the profile information
+        profile_url = "https://public.api.bsky.app/xrpc/app.bsky.actor.getProfile"
+        profile_params = {"actor": user_did}
+        
+        profile_response = requests.get(profile_url, params=profile_params)
+        
+        if profile_response.status_code != 200:
+            return f"Could not fetch profile for @{handle}. Status code: {profile_response.status_code}"
+        
+        profile_data = profile_response.json()
+        display_name = profile_data.get("displayName", handle)
+        
+        # Step 3: Get the author's feed
+        feed_url = "https://public.api.bsky.app/xrpc/app.bsky.feed.getAuthorFeed"
+        feed_params = {"actor": user_did, "limit": 1}
+        
+        feed_response = requests.get(feed_url, params=feed_params)
         
         if feed_response.status_code != 200:
-            return f"Error: Could not retrieve posts. Status code: {feed_response.status_code}"
-            
+            return f"Could not fetch posts for @{handle}. Status code: {feed_response.status_code}"
+        
         feed_data = feed_response.json()
         
-        # Look for the most recent post
-        if not feed_data or "feed" not in feed_data or len(feed_data["feed"]) == 0:
-            return f"@{handle} has no public posts on Bluesky"
-            
-        # Extract the latest post
+        # Check if there are any posts
+        if not feed_data.get("feed") or len(feed_data["feed"]) == 0:
+            return f"@{handle} has no posts on Bluesky"
+        
+        # Extract post information
         latest_post = feed_data["feed"][0]
-        if "text" in latest_post:
-            post_text = latest_post["text"]
-            post_time = latest_post.get("indexedAt", "unknown time")
-            
-            # Format the result
-            result = f"{matching_profile.get('displayName', handle)} (@{handle}) posted on {post_time}:\n\"{post_text}\""
-            return result
-        else:
-            return f"Could not extract text from the latest post by @{handle}"
+        post = latest_post.get("post", {})
+        record = post.get("record", {})
+        post_text = record.get("text", "")
+        post_time = record.get("createdAt", "unknown time")
         
+        # Format the result for the AI agent
+        result = f"{display_name} (@{handle}) posted on {post_time}:\n\"{post_text}\""
+        return result
+    
     except Exception as e:
         return f"Error fetching Bluesky post: {str(e)}"