Update chat_handler.py

chat_handler.py

@@ -384,5 +384,132 @@ async def chat(req: ChatRequest, x_session_id: str = Header(...)):
         traceback.print_exc()
         raise HTTPException(status_code=500, detail=str(e))
 
+async def handle_new_message(session: Session, user_input: str) -> str:
+    """Handle new message (not parameter followup) - for WebSocket"""
+    try:
+        # Get version config from session
+        version = session.get_version_config()
+        if not version:
+            log("❌ Version config not found")
+            return "Bir hata oluştu. Lütfen tekrar deneyin."
+        
+        # Get project config
+        project = next((p for p in cfg.projects if p.name == session.project_name), None)
+        if not project:
+            return "Proje konfigürasyonu bulunamadı."
+        
+        # Build intent detection prompt
+        prompt = build_intent_prompt(version, session.chat_history, project.default_locale)
+        
+        # Get LLM response
+        raw = await llm_generate(session, prompt, user_input)
+        
+        # Empty response fallback
+        if not raw:
+            log("⚠️ Empty response from LLM")
+            return "Üzgünüm, mesajınızı anlayamadım. Lütfen tekrar dener misiniz?"
+        
+        # Check for intent
+        intent_name, tail = _safe_intent_parse(raw)
+        
+        if intent_name:
+            # Find intent config
+            intent_config = next((i for i in version.intents if i.name == intent_name), None)
+            
+            if intent_config:
+                session.current_intent = intent_name
+                session.intent_config = intent_config
+                session.state = "collect_params"
+                log(f"🎯 Intent detected: {intent_name}")
+                
+                # Check if parameters were already extracted
+                if tail and _extract_parameters_from_response(tail, session, intent_config):
+                    log("📦 Some parameters extracted from initial response")
+                
+                # Check what parameters are missing
+                missing_params = [
+                    p for p in intent_config.parameters
+                    if p.required and p.variable_name not in session.variables
+                ]
+                
+                if not missing_params:
+                    # All required parameters collected, execute API
+                    return await _execute_api_call(session, intent_config)
+                else:
+                    # Need to collect more parameters
+                    param_prompt = build_parameter_prompt(
+                        intent_config, 
+                        session.variables, 
+                        session.chat_history,
+                        project.default_locale
+                    )
+                    param_question = await llm_generate(session, param_prompt, user_input)
+                    return _trim_response(param_question)
+        
+        # No intent detected, return general response
+        return _trim_response(raw)
+        
+    except Exception as e:
+        log(f"❌ Error in handle_new_message: {e}")
+        return "Bir hata oluştu. Lütfen tekrar deneyin."
+
+async def handle_parameter_followup(session: Session, user_input: str) -> str:
+    """Handle parameter collection followup - for WebSocket"""
+    try:
+        if not session.intent_config:
+            log("⚠️ No intent config in session")
+            session.reset_flow()
+            return "Üzgünüm, hangi işlem için bilgi istediğimi unuttum. Baştan başlayalım."
+        
+        intent_config = session.intent_config
+        
+        # Get project config
+        project = next((p for p in cfg.projects if p.name == session.project_name), None)
+        if not project:
+            return "Proje konfigürasyonu bulunamadı."
+        
+        # Try to extract parameters from user message
+        param_prompt = f"""
+        Extract parameters from user message: "{user_input}"
+        
+        Expected parameters:
+        {json.dumps([{
+            'name': p.name,
+            'type': p.type,
+            'required': p.required,
+            'extraction_prompt': p.extraction_prompt
+        } for p in intent_config.parameters if p.variable_name not in session.variables], ensure_ascii=False)}
+        
+        Return as JSON object with parameter names as keys.
+        """
+        
+        raw = await llm_generate(session, param_prompt, user_input)
+        _extract_parameters_from_response(raw, session, intent_config)
+        
+        # Check what parameters are still missing
+        missing_params = [
+            p for p in intent_config.parameters
+            if p.required and p.variable_name not in session.variables
+        ]
+        
+        if not missing_params:
+            # All parameters collected, execute API
+            return await _execute_api_call(session, intent_config)
+        else:
+            # Still need more parameters
+            param_prompt = build_parameter_prompt(
+                intent_config, 
+                session.variables, 
+                session.chat_history,
+                project.default_locale
+            )
+            param_question = await llm_generate(session, param_prompt, user_input)
+            return _trim_response(param_question)
+            
+    except Exception as e:
+        log(f"❌ Error in handle_parameter_followup: {e}")
+        session.reset_flow()
+        return "Bir hata oluştu. Lütfen tekrar deneyin."
+
 # Initialize LLM on module load
 setup_llm_provider()