feat: Add extract_topic and voice_chat tools to MCP service, enhancing functionality for topic extraction and chatbot responses. Update models and routes to support new features.

models/mcp_models.py

@@ -15,7 +15,7 @@ class ToolCallRequest(BaseModel):
         }
     )
     
-    tool_name: str = Field(..., description="Name of the MCP tool to call (e.g., 'detect_stance', 'match_keypoint_argument', 'transcribe_audio', 'generate_speech', 'generate_argument')")
+    tool_name: str = Field(..., description="Name of the MCP tool to call (e.g., 'detect_stance', 'match_keypoint_argument', 'transcribe_audio', 'generate_speech', 'generate_argument', 'extract_topic', 'voice_chat')")
     arguments: Dict[str, Any] = Field(default_factory=dict, description="Arguments for the tool (varies by tool)")
 
 class ToolCallResponse(BaseModel):
@@ -105,6 +105,40 @@ class GenerateSpeechResponse(BaseModel):
     
     audio_path: str = Field(..., description="Path to generated audio file")
 
+class ExtractTopicResponse(BaseModel):
+    """Response model for topic extraction"""
+    model_config = ConfigDict(
+        json_schema_extra={
+            "example": {
+                "text": "Governments should subsidize electric cars to encourage adoption.",
+                "topic": "government subsidies for electric vehicle adoption",
+                "timestamp": "2024-01-01T12:00:00"
+            }
+        }
+    )
+    
+    text: str = Field(..., description="The input text")
+    topic: str = Field(..., description="The extracted topic")
+    timestamp: Optional[str] = Field(None, description="Timestamp of extraction")
+
+class VoiceChatResponse(BaseModel):
+    """Response model for voice chat"""
+    model_config = ConfigDict(
+        json_schema_extra={
+            "example": {
+                "user_input": "What is climate change?",
+                "conversation_id": "uuid-here",
+                "response": "Climate change refers to long-term changes in global temperatures and weather patterns.",
+                "timestamp": "2024-01-01T12:00:00"
+            }
+        }
+    )
+    
+    user_input: str = Field(..., description="The user's input text")
+    conversation_id: Optional[str] = Field(None, description="The conversation ID")
+    response: str = Field(..., description="The chatbot's response")
+    timestamp: Optional[str] = Field(None, description="Timestamp of response")
+
 class ResourceInfo(BaseModel):
     """Information about an MCP resource"""
     uri: str

routes/mcp_routes.py

@@ -14,6 +14,8 @@ from services.mcp_service import mcp_server
 from services.stance_model_manager import stance_model_manager
 from services.label_model_manager import kpa_model_manager
 from services.generate_model_manager import generate_model_manager
+from services.topic_service import topic_service
+from services.chat_service import generate_chat_response
 from models.mcp_models import (
     ToolListResponse, 
     ToolInfo, 
@@ -22,7 +24,9 @@ from models.mcp_models import (
     DetectStanceResponse,
     MatchKeypointResponse,
     TranscribeAudioResponse,
-    GenerateSpeechResponse
+    GenerateSpeechResponse,
+    ExtractTopicResponse,
+    VoiceChatResponse
 )
 from models.generate import GenerateRequest, GenerateResponse
 from datetime import datetime
@@ -75,6 +79,30 @@ class GenerateSpeechRequest(BaseModel):
             }
         }
 
+class ExtractTopicRequest(BaseModel):
+    """Request pour extraire un topic d'un texte"""
+    text: str = Field(..., min_length=5, max_length=5000, description="Le texte/argument à partir duquel extraire le topic")
+    
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "text": "Governments should subsidize electric cars to encourage adoption."
+            }
+        }
+
+class VoiceChatRequest(BaseModel):
+    """Request pour générer une réponse de chatbot vocal"""
+    user_input: str = Field(..., description="L'entrée utilisateur (en anglais)")
+    conversation_id: Optional[str] = Field(None, description="ID de conversation pour maintenir le contexte")
+    
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "user_input": "What is climate change?",
+                "conversation_id": "optional-conversation-id"
+            }
+        }
+
 
 
 # ===== Routes MCP =====
@@ -90,6 +118,8 @@ async def mcp_health():
             "transcribe_audio",
             "generate_speech",
             "generate_argument",
+            "extract_topic",
+            "voice_chat",
             "health_check"
         ]
         return {
@@ -167,6 +197,29 @@ async def list_mcp_tools():
                     "required": ["topic", "position"]
                 }
             ),
+            ToolInfo(
+                name="extract_topic",
+                description="Extrait un topic à partir d'un texte/argument donné",
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "text": {"type": "string", "description": "Le texte/argument à partir duquel extraire le topic"}
+                    },
+                    "required": ["text"]
+                }
+            ),
+            ToolInfo(
+                name="voice_chat",
+                description="Génère une réponse de chatbot vocal en anglais",
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "user_input": {"type": "string", "description": "L'entrée utilisateur (en anglais)"},
+                        "conversation_id": {"type": "string", "description": "ID de conversation pour maintenir le contexte (optionnel)"}
+                    },
+                    "required": ["user_input"]
+                }
+            ),
             ToolInfo(
                 name="health_check",
                 description="Health check pour le serveur MCP",
@@ -244,6 +297,27 @@ async def call_mcp_tool(request: ToolCallRequest):
       }
     }
     ```
+    
+    6. **extract_topic** - Extraire un topic d'un texte:
+    ```json
+    {
+      "tool_name": "extract_topic",
+      "arguments": {
+        "text": "Governments should subsidize electric cars to encourage adoption."
+      }
+    }
+    ```
+    
+    7. **voice_chat** - Générer une réponse de chatbot vocal:
+    ```json
+    {
+      "tool_name": "voice_chat",
+      "arguments": {
+        "user_input": "What is climate change?",
+        "conversation_id": "optional-conversation-id"
+      }
+    }
+    ```
     """
     try:
         result = await mcp_server.call_tool(request.tool_name, request.arguments)
@@ -510,6 +584,62 @@ async def mcp_generate_argument(request: GenerateRequest):
         logger.error(f"Error in generate_argument: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Error executing tool generate_argument: {e}")
 
+@router.post("/tools/extract-topic", response_model=ExtractTopicResponse, summary="Extraire un topic d'un texte")
+async def mcp_extract_topic(request: ExtractTopicRequest):
+    """Extrait un topic à partir d'un texte/argument donné"""
+    try:
+        # Vérifier que le service est initialisé
+        if not topic_service.initialized:
+            topic_service.initialize()
+        
+        # Appeler directement le service (plus fiable que via MCP)
+        topic_text = topic_service.extract_topic(request.text)
+        
+        # Construire la réponse structurée
+        response = ExtractTopicResponse(
+            text=request.text,
+            topic=topic_text,
+            timestamp=datetime.now().isoformat()
+        )
+        
+        logger.info(f"Topic extracted from text '{request.text[:50]}...': {topic_text[:50]}...")
+        return response
+        
+    except ValueError as e:
+        logger.error(f"Validation error in extract_topic: {str(e)}")
+        raise HTTPException(status_code=400, detail=str(e))
+    except Exception as e:
+        logger.error(f"Error in extract_topic: {e}", exc_info=True)
+        raise HTTPException(status_code=500, detail=f"Error executing tool extract_topic: {e}")
+
+@router.post("/tools/voice-chat", response_model=VoiceChatResponse, summary="Générer une réponse de chatbot vocal")
+async def mcp_voice_chat(request: VoiceChatRequest):
+    """Génère une réponse de chatbot vocal en anglais"""
+    try:
+        # Appeler directement le service (plus fiable que via MCP)
+        response_text = generate_chat_response(
+            user_input=request.user_input,
+            conversation_id=request.conversation_id
+        )
+        
+        # Construire la réponse structurée
+        response = VoiceChatResponse(
+            user_input=request.user_input,
+            conversation_id=request.conversation_id,
+            response=response_text,
+            timestamp=datetime.now().isoformat()
+        )
+        
+        logger.info(f"Voice chat response generated for input '{request.user_input[:50]}...': {response_text[:50]}...")
+        return response
+        
+    except ValueError as e:
+        logger.error(f"Validation error in voice_chat: {str(e)}")
+        raise HTTPException(status_code=400, detail=str(e))
+    except Exception as e:
+        logger.error(f"Error in voice_chat: {e}", exc_info=True)
+        raise HTTPException(status_code=500, detail=f"Error executing tool voice_chat: {e}")
+
 @router.get("/tools/health-check", summary="Health check MCP (outil)")
 async def mcp_tool_health_check() -> Dict[str, Any]:
     """Health check via l'outil MCP"""

services/mcp_service.py

@@ -1,7 +1,7 @@
 """Service pour initialiser le serveur MCP avec FastMCP"""
 
 from mcp.server.fastmcp import FastMCP
-from typing import Dict, Any
+from typing import Dict, Any, Optional
 import logging
 
 from fastapi import FastAPI
@@ -11,6 +11,8 @@ from services.label_model_manager import kpa_model_manager
 from services.stt_service import speech_to_text
 from services.tts_service import text_to_speech
 from services.generate_model_manager import generate_model_manager
+from services.topic_service import topic_service
+from services.chat_service import generate_chat_response
 
 logger = logging.getLogger(__name__)
 
@@ -62,6 +64,30 @@ def generate_argument(topic: str, position: str) -> Dict[str, Any]:
         "argument": argument
     }
 
+@mcp_server.tool()
+def extract_topic(text: str) -> Dict[str, Any]:
+    """Extract a topic from the given text/argument"""
+    if not topic_service.initialized:
+        topic_service.initialize()
+    topic = topic_service.extract_topic(text)
+    return {
+        "text": text,
+        "topic": topic
+    }
+
+@mcp_server.tool()
+def voice_chat(user_input: str, conversation_id: Optional[str] = None) -> Dict[str, Any]:
+    """Generate a chatbot response for voice chat (English only)"""
+    response_text = generate_chat_response(
+        user_input=user_input,
+        conversation_id=conversation_id
+    )
+    return {
+        "user_input": user_input,
+        "conversation_id": conversation_id,
+        "response": response_text
+    }
+
 @mcp_server.resource("debate://prompt")
 def get_debate_prompt() -> str:
     return "Tu es un expert en débat. Génère 3 arguments PRO pour le topic donné. Sois concis et persuasif."
@@ -78,6 +104,8 @@ def health_check() -> Dict[str, Any]:
             "transcribe_audio",
             "generate_speech",
             "generate_argument",
+            "extract_topic",
+            "voice_chat",
             "health_check"
         ]
     except Exception: