Update agent.py

agent.py

@@ -300,73 +300,47 @@ class MagAgent:
     
     
     
-        class DownloadImageTool(Tool):
-            name = "download_chess_image"
-            description = "Downloads chess position image from task ID"
-            inputs = {'task_id': {'type': 'string'}}
-            output_type = "string"
-    
-            def forward(self, task_id: str) -> str:
-                try:
-                    response = requests.get(
-                        f"https://agents-course-unit4-scoring.hf.space/files/{task_id}",
-                        stream=True
-                    )
-                    response.raise_for_status()
-                    
-                    img_path = f"chess_{task_id}.png"
-                    with open(img_path, "wb") as f:
-                        for chunk in response.iter_content(8192):
-                            f.write(chunk)
-                    return img_path
-                except Exception as e:
-                    raise RuntimeError(f"Image download failed: {str(e)}")
-    
-    
-        
-        class ChessEngineTool(Tool):
-            import chess
-            import chess.engine
-            name = "stockfish_analysis"
-            description = "Analyzes chess position using Stockfish"
-            inputs = {'fen': {'type': 'string'}}
-            output_type = "string"
-    
-            def forward(self, fen: str) -> str:
-                try:
-                    board = chess.Board(fen)
-                    engine = chess.engine.SimpleEngine.popen_uci("stockfish")
-                    result = engine.play(board, chess.engine.Limit(time=2.0))
-                    engine.quit()
-                    return board.san(result.move)
-                except Exception as e:
-                    return f"Engine error: {str(e)}"
-    
-        async def analyze_position(self, task_id: str):
+    class DownloadImageTool(Tool):
+        name = "download_chess_image"
+        description = "Downloads chess position image from task ID"
+        inputs = {'task_id': {'type': 'string'}}
+        output_type = "string"
+
+        def forward(self, task_id: str) -> str:
             try:
-                # Step 1: Download image
-                img_path = await self.tools[0](task_id)
-                
-                # Step 2: Get multimodal analysis
-                response = await self.model.acreate(
-                    messages=[{
-                        "role": "user",
-                        "content": [
-                            {"type": "text", "text": """Analyze this chess position. 
-                            It's black's turn. Provide the winning move in algebraic notation.
-                            Respond ONLY with the move, nothing else."""},
-                            {"type": "image_url", "image_url": {"url": f"file://{img_path}"}}
-                        ]
-                    }],
-                    temperature=0.1
+                response = requests.get(
+                    f"https://agents-course-unit4-scoring.hf.space/files/{task_id}",
+                    stream=True
                 )
+                response.raise_for_status()
                 
-                return response.choices[0].message.content
-                
+                img_path = f"chess_{task_id}.png"
+                with open(img_path, "wb") as f:
+                    for chunk in response.iter_content(8192):
+                        f.write(chunk)
+                return img_path
             except Exception as e:
-                return f"Analysis failed: {str(e)}"
-    
+                raise RuntimeError(f"Image download failed: {str(e)}")
+
     
+    class ChessEngineTool(Tool):
+        import chess
+        import chess.engine
+        name = "stockfish_analysis"
+        description = "Analyzes chess position using Stockfish"
+        inputs = {'fen': {'type': 'string'}}
+        output_type = "string"
+
+        def forward(self, fen: str) -> str:
+            try:
+                board = chess.Board(fen)
+                engine = chess.engine.SimpleEngine.popen_uci("stockfish")
+                result = engine.play(board, chess.engine.Limit(time=2.0))
+                engine.quit()
+                return board.san(result.move)
+            except Exception as e:
+                return f"Engine error: {str(e)}"
+        
     
             
     class PythonCodeReaderTool(Tool):