diff --git a/api/app.py b/api/app.py
index c315844e..c1ceb7a3 100644
--- a/api/app.py
+++ b/api/app.py
@@ -58,7 +58,10 @@ def chat():
     chat_history = data.get('history', [])
     api_key = data.get('apikey')
 
+
     if df is not None:
+        
+        print(df.head(1))
         documents = split_dataframe_into_documents(df)
         
         embeddings = OpenAIEmbeddings(api_key=api_key)
@@ -74,27 +77,24 @@ def chat():
 
         return jsonify(reply=bot_reply)
     else:
-        print("No df")
-        return jsonify(reply="No data available."), 400
-
-    # messages = [{"role": "system", "content": "You are a helpful assistant."}]
-    # for msg in chat_history:
-    #     messages.append({"role": "user" if msg['origin'] == 'user' else "assistant", "content": msg['text']})
-    # messages.append({"role": "user", "content": user_message})
-
-    # try:
-    #     client = openai.OpenAI(api_key = api_key)
-    #     response = client.chat.completions.create(
-    #         model="gpt-3.5-turbo",
-    #         messages=messages,
-    #         max_tokens=200
-    #     )
-    #     bot_reply = response.choices[0].message.content.strip()
-
-    #     return jsonify(reply=bot_reply)
-    # except Exception as e:
-    #     print(f"Error: {e}")
-    #     return jsonify(reply="Desculpe, ocorreu um erro ao processar sua mensagem."), 500
+        messages = [{"role": "system", "content": "You are a helpful assistant."}]
+        for msg in chat_history:
+            messages.append({"role": "user" if msg['origin'] == 'user' else "assistant", "content": msg['text']})
+        messages.append({"role": "user", "content": user_message})
+
+        try:
+            client = openai.OpenAI(api_key = api_key)
+            response = client.chat.completions.create(
+                model="gpt-3.5-turbo",
+                messages=messages,
+                max_tokens=200
+            )
+            bot_reply = response.choices[0].message.content.strip()
+
+            return jsonify(reply=bot_reply)
+        except Exception as e:
+            print(f"Error: {e}")
+            return jsonify(reply="Desculpe, ocorreu um erro ao processar sua mensagem."), 500
 
 
 def shutdown_server():
@@ -116,7 +116,7 @@ def receive_file():
     if file.filename == '':
         return jsonify({'error': 'No selected file'}), 400
     if file:
-        df = pd.read_csv(file)
+        df = pd.read_csv(file, on_bad_lines='skip')
         return jsonify({'message': 'File uploaded successfully'}), 200
 
 
@@ -282,40 +282,6 @@ def apikey():
         return jsonify(reply="Desculpe, ocorreu um erro ao processar sua mensagem."), 500
 
 
-# @app.route('/chat', methods=['POST'])
-# def chat():
-#     global df
-#     if df is not None:
-#         # run rag
-#         print(df.head(1))
-#     else:
-#         print("No df")
-#     data = request.get_json()
-#     user_message = data.get('message')
-#     chat_history = data.get('history', [])
-#     api_key = data.get('apikey')
-
-#     messages = [{"role": "system", "content": "You are a helpful assistant."}]
-#     for msg in chat_history:
-#         messages.append({"role": "user" if msg['origin'] == 'user' else "assistant", "content": msg['text']})
-#     messages.append({"role": "user", "content": user_message})
-
-#     try:
-#         client = openai.OpenAI(api_key = api_key)
-#         response = client.chat.completions.create(
-#             model="gpt-3.5-turbo", # ou a gente poderia ver com gpt 4 mas por enquanto coloquei 3.5
-#             messages=messages,
-#             max_tokens=200
-#         )
-#         bot_reply = response.choices[0].message.content.strip()
-
-#         return jsonify(reply=bot_reply)
-#     except Exception as e:
-#         print(f"Error: {e}")
-#         return jsonify(reply="Desculpe, ocorreu um erro ao processar sua mensagem."), 500
-
-
-
 if __name__ == '__main__':
     training_progress = {
         'training_progress': 0,
diff --git a/src/components/chatbot/chatbot.tsx b/src/components/chatbot/chatbot.tsx
index c3ab198e..52ac255f 100644
--- a/src/components/chatbot/chatbot.tsx
+++ b/src/components/chatbot/chatbot.tsx
@@ -50,8 +50,9 @@ const ChatBot: React.FC = () => {
             const errorMessage: Message = { text: "Desculpe, ocorreu um erro. Tente novamente.", origin: 'bot' };
             setChatHistory(prevHistory => [...prevHistory, errorMessage]);
         }
-
+        console.log(message)
         setMessage("");
+        console.log(message)
     };
 
     const sendAPIKeyMessage = () => {
@@ -130,7 +131,7 @@ const ChatBot: React.FC = () => {
                                 value={message}
                                 onChange={(e) => setMessage(e.target.value)}
                                 onKeyPress={(e) => {
-                                    if (e.key === 'Enter') sendMessage();
+                                    if (e.key === 'Enter') sendMessage()
                                 }}
                             />
                             <button
diff --git a/src/pages/views/trainView.tsx b/src/pages/views/trainView.tsx
index 6d55b19e..f64c3475 100644
--- a/src/pages/views/trainView.tsx
+++ b/src/pages/views/trainView.tsx
@@ -19,6 +19,7 @@ export default function TrainView() {
     const [hasTrained, setHasTrained] = useState(false);
     const [isCancelling, setIsCancelling] = useState(false);
     const [trainingCompleted, setTrainingCompleted] = useState(false);
+    const [hasCancelled, setHasCancelled] = useState(false);
 
     const handleChangeSelectedColumn = (event: React.ChangeEvent<HTMLSelectElement>) => {
         setSelectedColumn(Number(event.target.value));
@@ -30,8 +31,9 @@ export default function TrainView() {
 
     const handleCancelTraining = async () => {
         setIsCancelling(true);
+        setHasCancelled(true);
         try {
-            await axios.post('http://localhost:5000/cancel-training');
+            await axios.post('http://localhost:5000/cancel-training');    
             alert('Treinamento cancelado com sucesso!');
         } catch (error) {
             console.error('Erro ao cancelar o treinamento:', error);
@@ -44,6 +46,7 @@ export default function TrainView() {
         setIsLoading(true);
         setHasTrained(true);
         setTrainingCompleted(false);
+        setHasCancelled(false);
         setLoadingProgress(0);
         setTrainLosses([]);
         setValidLosses([]);
@@ -85,6 +88,7 @@ export default function TrainView() {
 
     const handleNbSubmit = async () => {
         setIsLoading(true);
+        setHasCancelled(false);
         setTrainingCompleted(false);
         setLoadingProgress(0);
 
@@ -213,7 +217,7 @@ export default function TrainView() {
     }, []);
 
     useEffect(() => {
-        if (trainingCompleted) {
+        if (trainingCompleted && !hasCancelled) {
             toast("Seu modelo está disponível para uso na aba de Classificação", {
                 position: "bottom-right",
                 autoClose: 5000,
diff --git a/yarn.lock b/yarn.lock
index f15d9456..cecaf240 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -9902,7 +9902,7 @@ react-dev-utils@^12.0.0:
     strip-ansi "^6.0.1"
     text-table "^0.2.0"
 
-"react-dom@^17.0.0 || ^18.0.0", react-dom@^18.0.0, react-dom@^18.2.0, react-dom@>=16.6.0, react-dom@>=16.8, react-dom@>=16.8.0, react-dom@>=18:
+"react-dom@^16.0.0 || ^17.0.0 || ^18.0.0", "react-dom@^17.0.0 || ^18.0.0", react-dom@^18.0.0, react-dom@^18.2.0, react-dom@>=16.6.0, react-dom@>=16.8, react-dom@>=16.8.0, react-dom@>=18:
   version "18.2.0"
   resolved "https://registry.npmjs.org/react-dom/-/react-dom-18.2.0.tgz"
   integrity sha512-6IMTriUmvsjHUjNtEDudZfuDQUoWXVxKHhlEGSk81n4YFS+r/Kl99wXiwlVXtPBtJenozv2P+hxDsw9eA7Xo6g==
@@ -10023,6 +10023,11 @@ react-scripts@5.0.0:
   optionalDependencies:
     fsevents "^2.3.2"
 
+react-spinners@^0.13.8:
+  version "0.13.8"
+  resolved "https://registry.npmjs.org/react-spinners/-/react-spinners-0.13.8.tgz"
+  integrity sha512-3e+k56lUkPj0vb5NDXPVFAOkPC//XyhKPJjvcGjyMNPWsBKpplfeyialP74G7H7+It7KzhtET+MvGqbKgAqpZA==
+
 react-toastify@^10.0.5:
   version "10.0.5"
   resolved "https://registry.npmjs.org/react-toastify/-/react-toastify-10.0.5.tgz"
@@ -10040,7 +10045,7 @@ react-transition-group@^4.4.5:
     loose-envify "^1.4.0"
     prop-types "^15.6.2"
 
-react@*, "react@^17.0.0 || ^18.0.0", react@^18.0.0, react@^18.2.0, "react@>= 16", react@>=0.13, react@>=16, react@>=16.6.0, react@>=16.8, react@>=16.8.0, react@>=18:
+react@*, "react@^16.0.0 || ^17.0.0 || ^18.0.0", "react@^17.0.0 || ^18.0.0", react@^18.0.0, react@^18.2.0, "react@>= 16", react@>=0.13, react@>=16, react@>=16.6.0, react@>=16.8, react@>=16.8.0, react@>=18:
   version "18.2.0"
   resolved "https://registry.npmjs.org/react/-/react-18.2.0.tgz"
   integrity sha512-/3IjMdb2L9QbBdWiW5e3P2/npwMBaU9mHCSCUzNln0ZCYbcfTsGbTJrU/kGemdH2IWmB2ioZ+zkxtmq6g09fGQ==