init

internal/handler/ai_handler.go

@@ -0,0 +1,147 @@
+package handler
+
+import (
+	"net/http"
+
+	"accounting-app/internal/service"
+
+	"github.com/gin-gonic/gin"
+)
+
+// AIHandler handles AI bookkeeping API requests
+type AIHandler struct {
+	aiService *service.AIBookkeepingService
+}
+
+// NewAIHandler creates a new AIHandler
+func NewAIHandler(aiService *service.AIBookkeepingService) *AIHandler {
+	return &AIHandler{
+		aiService: aiService,
+	}
+}
+
+// ChatRequest represents a chat request
+type ChatRequest struct {
+	SessionID string `json:"session_id"`
+	Message   string `json:"message" binding:"required"`
+}
+
+// TranscribeRequest represents a transcription request
+type TranscribeRequest struct {
+	// Audio file is sent as multipart form data
+}
+
+// ConfirmRequest represents a transaction confirmation request
+type ConfirmRequest struct {
+	SessionID string `json:"session_id" binding:"required"`
+}
+
+// RegisterRoutes registers AI routes
+func (h *AIHandler) RegisterRoutes(rg *gin.RouterGroup) {
+	ai := rg.Group("/ai")
+	{
+		ai.POST("/chat", h.Chat)
+		ai.POST("/transcribe", h.Transcribe)
+		ai.POST("/confirm", h.Confirm)
+	}
+}
+
+// Chat handles chat messages for AI bookkeeping
+// POST /api/v1/ai/chat
+// Requirements: 12.1, 12.5
+func (h *AIHandler) Chat(c *gin.Context) {
+	var req ChatRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"success": false,
+			"error":   "Invalid request: " + err.Error(),
+		})
+		return
+	}
+
+	// Get user ID from context (default to 1 for now)
+	userID := uint(1)
+	if id, exists := c.Get("user_id"); exists {
+		userID = id.(uint)
+	}
+
+	response, err := h.aiService.ProcessChat(c.Request.Context(), userID, req.SessionID, req.Message)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"success": false,
+			"error":   "Failed to process chat: " + err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"data":    response,
+	})
+}
+
+// Transcribe handles audio transcription
+// POST /api/v1/ai/transcribe
+// Requirements: 12.2, 12.6
+func (h *AIHandler) Transcribe(c *gin.Context) {
+	// Get audio file from form
+	file, header, err := c.Request.FormFile("audio")
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"success": false,
+			"error":   "No audio file provided: " + err.Error(),
+		})
+		return
+	}
+	defer file.Close()
+
+	// Transcribe audio
+	result, err := h.aiService.TranscribeAudio(c.Request.Context(), file, header.Filename)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"success": false,
+			"error":   "Failed to transcribe audio: " + err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"data":    result,
+	})
+}
+
+// Confirm handles transaction confirmation
+// POST /api/v1/ai/confirm
+// Requirements: 12.3, 12.7, 12.8
+func (h *AIHandler) Confirm(c *gin.Context) {
+	var req ConfirmRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{
+			"success": false,
+			"error":   "Invalid request: " + err.Error(),
+		})
+		return
+	}
+
+	// Get user ID from context (default to 1 for now)
+	userID := uint(1)
+	if id, exists := c.Get("user_id"); exists {
+		userID = id.(uint)
+	}
+
+	transaction, err := h.aiService.ConfirmTransaction(c.Request.Context(), req.SessionID, userID)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"success": false,
+			"error":   "Failed to confirm transaction: " + err.Error(),
+		})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"data":    transaction,
+		"message": "Transaction created successfully",
+	})
+}