Update app.py

app.py

@@ -1,25 +1,24 @@
-import PyPDF2
 import os
-from bs4 import BeautifulSoup
-import tempfile
 import csv
 import json
-import xml.etree.ElementTree as ET
 import docx
 import pptx
-import openpyxl
 import re
 import nltk
 import time
+import PyPDF2
+import tempfile
+import openpyxl
 import requests
 import gradio as gr
+from bs4 import BeautifulSoup
+import xml.etree.ElementTree as ET
 from nltk.tokenize import word_tokenize
-from langchain_community.embeddings import SentenceTransformerEmbeddings
 from langchain_community.vectorstores import FAISS
-from langchain_community.llms import HuggingFaceEndpoint
-from langchain.schema import SystemMessage, HumanMessage, AIMessage
-from langchain_community.chat_models.huggingface import ChatHuggingFace
 from youtube_transcript_api import YouTubeTranscriptApi
+from langchain.schema import SystemMessage, HumanMessage, AIMessage
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_community.embeddings import SentenceTransformerEmbeddings
 from youtube_transcript_api._errors import NoTranscriptFound, TranscriptsDisabled, VideoUnavailable
 nltk.download('punkt')
 nltk.download('omw-1.4')
@@ -203,7 +202,7 @@ def Chat_Message(history):
     message=HumanMessage(content=history[-1][0])
     messages.append(message)
     response = chat_model.invoke(messages)
-    messages.append(response.content)
+    messages.append(AIMessage(content=response.content))
 
     if len(messages) >= 8:
       messages = messages[-8:]
@@ -239,7 +238,7 @@ def Web_Search(history):
     msg==HumanMessage(content=augmented_prompt)
     messages.append(msg)
     response = chat_model.invoke(msg)
-    messages.append(response.content)
+    messages.append(AIMessage(content=response.content))
 
     if len(messages) >= 8:
       messages = messages[-8:]
@@ -269,7 +268,7 @@ def Chart_Generator(history):
         messages.append(prompt)
 
         res = chat_model.invoke(messages)
-        messages.append(res.content)
+        messages.append(AIMessage(content=response.content))
 
         if len(messages) >= 8:
             messages = messages[-8:]
@@ -322,7 +321,7 @@ def Link_Scratch(history):
             message = HumanMessage(content=augmented_prompt)
             messages.append(message)
             response = chat_model.invoke(messages)
-            messages.append(response.content)
+            messages.append(AIMessage(content=response.content))
 
             if len(messages) >= 1:
                 messages = messages[-1:]
@@ -381,7 +380,7 @@ def File_Interact(history,filepath):
             message = HumanMessage(content=augmented_prompt)
             messages.append(message)
             response = chat_model.invoke(messages)
-            messages.append(response.content)
+            messages.append(AIMessage(content=response.content))
 
             if len(messages) >= 1:
                 messages = messages[-1:]