website/src/components/ChatInterface.jsx

import { useState, useRef, useCallback, useEffect } from 'react';
import styled from 'styled-components';
import axios from 'axios';
import { API_URL } from '../config';

const api = axios.create({
  baseURL: API_URL,
  timeout: 10000,
});

const ChatWrapper = styled.div`
  display: flex;
  flex-direction: column;
  height: ${({ $compact }) => ($compact ? '560px' : '70vh')};
  max-width: 900px;
  width: 100%;
  margin: 0 auto;
  background-color: #fff;
  border: 1px solid rgba(0, 0, 0, 0.08);
  border-radius: 8px;
  overflow: hidden;

  @media (max-width: 620px) {
    height: 560px;
  }
`;

const ChatContainer = styled.div`
  flex-grow: 1;
  overflow-y: auto;
  padding: 20px;
  display: flex;
  flex-direction: column;
  gap: 12px;
  background: #f1efe8;
`;

const Message = styled.div`
  padding: 12px 16px;
  border-radius: 8px;
  max-width: 80%;
  word-break: break-word;
  background-color: ${({ sender }) => (sender === 'user' ? '#378add' : '#ffffff')};
  color: ${({ sender }) => (sender === 'user' ? '#fff' : '#2f3540')};
  align-self: ${({ sender }) => (sender === 'user' ? 'flex-end' : 'flex-start')};
  border: ${({ sender }) => (sender === 'user' ? '0' : '1px solid rgba(0, 0, 0, 0.08)')};
  line-height: 1.55;
`;

const InputArea = styled.div`
  position: sticky;
  bottom: 0;
  background-color: #fff;
  padding: 16px;
  border-top: 1px solid rgba(0, 0, 0, 0.08);
`;

const InputContainer = styled.div`
  display: flex;
  align-items: center;
  gap: 10px;
`;

const InputField = styled.input`
  flex-grow: 1;
  padding: 12px 16px;
  min-height: 54px;
  border: 1px solid #b8c2cc;
  border-radius: 8px;
  font-size: 16px;
  outline: none;
  transition: border-color 0.2s ease;

  &:focus {
    border-color: #378add;
  }
`;

const SendButton = styled.button`
  min-width: 58px;
  min-height: 54px;
  padding: 0 18px;
  background-color: #378add;
  color: #fff;
  border: none;
  border-radius: 8px;
  font-size: 16px;
  cursor: pointer;
  transition: background-color 0.2s ease;

  &:hover {
    background-color: #185fa5;
  }

  &:disabled {
    background-color: #b0bec5;
    cursor: not-allowed;
  }
`;

const EmptyState = styled.div`
  display: grid;
  gap: 14px;
  max-width: 520px;
  margin: 24px auto;
  color: #2f3540;
  font-size: 20px;
  line-height: 1.5;

  span {
    color: #185fa5;
    font-size: 15px;
    font-weight: 700;
  }
`;

const SuggestionRow = styled.div`
  display: flex;
  flex-wrap: wrap;
  gap: 8px;

  button {
    min-height: 38px;
    padding: 0 12px;
    border: 0.5px solid #85b7eb;
    border-radius: 999px;
    background: #ffffff;
    color: #185fa5;
    font-weight: 700;
    cursor: pointer;
  }
`;

const ChatInterface = ({ documentName, business, onMessageSaved, onConversationChange, compact = false }) => {
  const [messages, setMessages] = useState([]);
  const [input, setInput] = useState('');
  const [isLoading, setIsLoading] = useState(false);
  const [conversationId, setConversationId] = useState(null);
  const chatContainerRef = useRef(null);

  // Auto-scroll to bottom when messages change
  useEffect(() => {
    if (chatContainerRef.current) {
      chatContainerRef.current.scrollTop = chatContainerRef.current.scrollHeight;
    }
  }, [messages]);

  useEffect(() => {
    if (business?.id) {
      const stored = localStorage.getItem(`sahel_chat_history_${business.id}`);
      const storedConvId = localStorage.getItem(`sahel_chat_conv_id_${business.id}`);
      if (stored) {
        try {
          setMessages(JSON.parse(stored));
        } catch (e) {
          setMessages([]);
        }
      } else {
        setMessages([]);
      }
      setConversationId(storedConvId || null);
    } else {
      setMessages([]);
      setConversationId(null);
    }
    setInput('');
  }, [business?.id]);

  const sendMessage = useCallback(async () => {
    const trimmedInput = input.trim();
    if (!trimmedInput || isLoading || !business) return;

    const newUserMessage = { id: Date.now(), text: trimmedInput, sender: 'user' };
    setMessages((prev) => {
      const updated = [...prev, newUserMessage];
      localStorage.setItem(`sahel_chat_history_${business.id}`, JSON.stringify(updated));
      return updated;
    });
    setInput('');
    setIsLoading(true);

    try {
      const response = await api.post(`/businesses/${business.id}/chat`, {
        question: trimmedInput,
        conversation_id: conversationId,
      });
      const newConvId = response.data.conversation_id;
      setConversationId(newConvId);
      localStorage.setItem(`sahel_chat_conv_id_${business.id}`, newConvId);
      onConversationChange?.(newConvId);

      const newBotMessage = { id: Date.now(), text: response.data.answer, sender: 'bot' };
      setMessages((prev) => {
        const updated = [...prev, newBotMessage];
        localStorage.setItem(`sahel_chat_history_${business.id}`, JSON.stringify(updated));
        return updated;
      });
      onMessageSaved?.();
    } catch {
      const errorMsg = { id: Date.now(), text: 'Error: Could not get response', sender: 'bot' };
      setMessages((prev) => {
        const updated = [...prev, errorMsg];
        localStorage.setItem(`sahel_chat_history_${business.id}`, JSON.stringify(updated));
        return updated;
      });
    } finally {
      setIsLoading(false);
    }
  }, [business, conversationId, input, isLoading, onConversationChange, onMessageSaved]);

  return (
    <ChatWrapper $compact={compact}>
      <ChatContainer ref={chatContainerRef}>
        {messages.length === 0 ? (
          <EmptyState>
            <span>AI-POWERED CONCIERGE</span>
            <div>
              {documentName
                ? `Posez une question sur "${documentName}" pour tester ${business?.name}.`
                : 'Bonjour, je peux repondre aux questions, preparer une reservation ou expliquer les services disponibles.'}
            </div>
            <SuggestionRow>
              <button type="button" onClick={() => setInput('Quels sont vos services ?')}>
                Services
              </button>
              <button type="button" onClick={() => setInput('Quels sont vos tarifs ?')}>
                Tarifs
              </button>
            </SuggestionRow>
          </EmptyState>
        ) : (
          messages.map((message) => (
            <Message key={message.id} sender={message.sender}>
              {message.text}
            </Message>
          ))
        )}
      </ChatContainer>
      <InputArea>
        <InputContainer>
          <InputField
            type="text"
            placeholder="Votre question en arabe, francais ou darija..."
            value={input}
            onChange={(e) => setInput(e.target.value)}
            onKeyDown={(e) => e.key === 'Enter' && sendMessage()}
            disabled={isLoading}
          />
          <SendButton onClick={sendMessage} disabled={!input.trim() || isLoading || !business}>
            {isLoading ? '...' : '▷'}
          </SendButton>
        </InputContainer>
      </InputArea>
    </ChatWrapper>
  );
};

export default ChatInterface;