Update server.py

server.py

@@ -147,32 +147,80 @@ SOURCE:
 </source_material>
 """
 
-@mcp.tool
-async def analyze_file(ctx: Context, file: bytes) -> str:
+mcp = FastMCP("file-analyzer")
+
+# Helper functions for different file types
+async def process_pdf(file_bytes: bytes) -> str:
+    """Process PDF file and extract text/info"""
+    pdf_file = io.BytesIO(file_bytes)
+    reader = PdfReader(pdf_file)
+    
+    num_pages = len(reader.pages)
+    text = ""
+    for page in reader.pages:
+        text += page.extract_text()
+    
+    return f"PDF Analysis:\n- Pages: {num_pages}\n- Text length: {len(text)} chars\n\nFirst 500 chars:\n{text[:500]}"
+
+async def process_image(file_bytes: bytes) -> str:
+    """Process image file"""
+    img = Image.open(io.BytesIO(file_bytes))
+    return f"Image Analysis:\n- Format: {img.format}\n- Size: {img.size}\n- Mode: {img.mode}"
+
+async def process_text(file_bytes: bytes) -> str:
+    """Process text file"""
+    try:
+        text = file_bytes.decode('utf-8')
+        lines = text.split('\n')
+        words = len(text.split())
+        return f"Text Analysis:\n- Lines: {len(lines)}\n- Words: {words}\n- Characters: {len(text)}\n\nFirst 500 chars:\n{text[:500]}"
+    except UnicodeDecodeError:
+        return "Error: File is not valid UTF-8 text"
+
+@mcp.tool 
+async def analyze_file(
+    ctx: Context, 
+    file: bytes,
+    file_type: Optional[Literal["text", "image", "pdf"]] = None
+) -> str: 
     """
-    Example tool that:
-      1. Receives a file
-      2. If type is ambiguous, asks for clarification
-      3. Then returns a processed summary
+    Analyzes a file.
+    
+    Args:
+        file: The file content
+        file_type: Type of file (text/image/pdf). If not provided, will ask user.
     """
-
-    # 1) Basic file info
-    info = f"Received file of size {len(file)} bytes."
-
-    # 2) Elicit choice if needed
-    result = await ctx.elicit(
-        message="What is the file type (text/image/pdf)?",
-        response_type=["text", "image", "pdf"]
-    )
-
-    if result.action == "accept":
-        file_type = result.data
-    else:
-        return "Elicitation declined or cancelled"
-
-    # 3) Do some processing based on type
-    return f"{info} User said type: {file_type}"
-
+    # Basic info
+    info = f"Received file of size {len(file)} bytes.\n\n"
+    
+    # If file_type not provided, elicit it from user
+    if file_type is None:
+        result = await ctx.elicit(
+            message="What is the file type?",
+            response_type=["text", "image", "pdf"]
+        )
+        
+        if result.action == "accept":
+            file_type = result.data
+        else:
+            return "Elicitation cancelled by user"
+    
+    # Route to appropriate processor
+    try:
+        if file_type == "pdf":
+            result = await process_pdf(file)
+        elif file_type == "image":
+            result = await process_image(file)
+        elif file_type == "text":
+            result = await process_text(file)
+        else:
+            return info + f"Unknown file type: {file_type}"
+        
+        return info + result
+    
+    except Exception as e:
+        return info + f"Error processing {file_type}: {str(e)}"
+        
 if __name__ == "__main__":
     print("Starting FastMCP server on port 7860...")