feat: more tools

app.py

@@ -1,32 +1,64 @@
 import gradio as gr
 import requests
 import json
+import random
 from typing import Dict, Any, Optional
 
-def get_xkcd_comic(comic_id: str = "") -> str:
+def get_latest() -> str:
+    """
+    Fetch the latest XKCD comic.
+    
+    Returns:
+        str: JSON string containing latest comic information
+    """
+    try:
+        url = "https://xkcd.com/info.0.json"
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        
+        comic_data = response.json()
+        
+        formatted_response = {
+            "num": comic_data["num"],
+            "title": comic_data["title"],
+            "alt": comic_data["alt"],
+            "img": comic_data["img"],
+            "year": comic_data["year"],
+            "month": comic_data["month"],
+            "day": comic_data["day"],
+            "transcript": comic_data.get("transcript", ""),
+            "safe_title": comic_data["safe_title"]
+        }
+        
+        return json.dumps(formatted_response, indent=2)
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching latest comic: {str(e)}"
+    except KeyError as e:
+        return f"Error parsing comic data: Missing field {str(e)}"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"
+
+def get_comic(comic_id: str) -> str:
     """
-    Fetch XKCD comic information by ID or get the latest comic.
+    Fetch a specific XKCD comic by ID.
     
     Args:
-        comic_id (str): Comic ID number (leave empty for latest comic)
+        comic_id (str): Comic ID number
         
     Returns:
-        str: JSON string containing comic information including title, alt text, and image URL
+        str: JSON string containing comic information
     """
     try:
-        if comic_id.strip():
-            # Get specific comic by ID
-            url = f"https://xkcd.com/{comic_id.strip()}/info.0.json"
-        else:
-            # Get latest comic
-            url = "https://xkcd.com/info.0.json"
+        if not comic_id.strip():
+            return "Error: Comic ID is required"
             
+        url = f"https://xkcd.com/{comic_id.strip()}/info.0.json"
         response = requests.get(url, timeout=10)
         response.raise_for_status()
         
         comic_data = response.json()
         
-        # Format the response nicely
         formatted_response = {
             "num": comic_data["num"],
             "title": comic_data["title"],
@@ -42,7 +74,52 @@ def get_xkcd_comic(comic_id: str = "") -> str:
         return json.dumps(formatted_response, indent=2)
         
     except requests.exceptions.RequestException as e:
-        return f"Error fetching comic: {str(e)}"
+        return f"Error fetching comic {comic_id}: {str(e)}"
+    except KeyError as e:
+        return f"Error parsing comic data: Missing field {str(e)}"
+    except Exception as e:
+        return f"Unexpected error: {str(e)}"
+
+def get_random() -> str:
+    """
+    Fetch a random XKCD comic.
+    
+    Returns:
+        str: JSON string containing random comic information
+    """
+    try:
+        # First get the latest comic to know the range
+        latest_response = requests.get("https://xkcd.com/info.0.json", timeout=10)
+        latest_response.raise_for_status()
+        latest_num = latest_response.json()["num"]
+        
+        # Generate random comic ID (excluding 404 which doesn't exist)
+        comic_id = random.randint(1, latest_num)
+        if comic_id == 404:
+            comic_id = 405
+            
+        url = f"https://xkcd.com/{comic_id}/info.0.json"
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        
+        comic_data = response.json()
+        
+        formatted_response = {
+            "num": comic_data["num"],
+            "title": comic_data["title"],
+            "alt": comic_data["alt"],
+            "img": comic_data["img"],
+            "year": comic_data["year"],
+            "month": comic_data["month"],
+            "day": comic_data["day"],
+            "transcript": comic_data.get("transcript", ""),
+            "safe_title": comic_data["safe_title"]
+        }
+        
+        return json.dumps(formatted_response, indent=2)
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching random comic: {str(e)}"
     except KeyError as e:
         return f"Error parsing comic data: Missing field {str(e)}"
     except Exception as e:
@@ -104,23 +181,40 @@ def search_xkcd_transcript(search_term: str) -> str:
     except Exception as e:
         return f"Search error: {str(e)}"
 
+# Helper function for the Gradio interface
+def get_xkcd_comic(comic_id: str = "") -> str:
+    """Wrapper function for Gradio interface compatibility"""
+    if not comic_id.strip():
+        return get_latest()
+    else:
+        return get_comic(comic_id)
+
 # Create Gradio interface
 with gr.Blocks(title="XKCD MCP Server") as demo:
     gr.Markdown("# XKCD MCP Server")
     gr.Markdown("This server provides tools to fetch and search XKCD comics via MCP protocol.")
     
     with gr.Tab("Get Comic"):
-        comic_input = gr.Textbox(
-            label="Comic ID", 
-            placeholder="Leave empty for latest comic",
-            value=""
-        )
+        with gr.Row():
+            with gr.Column():
+                latest_btn = gr.Button("Get Latest Comic")
+                random_btn = gr.Button("Get Random Comic")
+            with gr.Column():
+                comic_input = gr.Textbox(
+                    label="Comic ID", 
+                    placeholder="Enter comic ID number",
+                    value=""
+                )
+                specific_btn = gr.Button("Get Specific Comic")
+        
         comic_output = gr.Textbox(
             label="Comic Data (JSON)",
             lines=15
         )
-        comic_btn = gr.Button("Get Comic")
-        comic_btn.click(get_xkcd_comic, inputs=[comic_input], outputs=[comic_output])
+        
+        latest_btn.click(get_latest, outputs=[comic_output])
+        random_btn.click(get_random, outputs=[comic_output])
+        specific_btn.click(get_comic, inputs=[comic_input], outputs=[comic_output])
     
     with gr.Tab("Search Comics"):
         search_input = gr.Textbox(