Update app.py

app.py

@@ -36,7 +36,6 @@ MODEL_DICT = {
 DATA_PATH = 'models/dataset.xlsx'
 EMBEDDINGS_PATH = 'models/embeddings.pkl'
 
-
 class ChatbotModel:
     def __init__(self, model=DEFAULT_MODEL):
         self._chatbot = Chatbot()
@@ -49,10 +48,6 @@ class ChatbotModel:
     def chat(self, question):
         return self._chatbot.answer_question(question)
 
-    def eval(self, model, predict_method):
-        return self._chatbot.eval(model_name=model, predict_method=predict_method)
-
-
 class Chatbot:
     def __init__(self):
         # Initialize variables
@@ -138,16 +133,6 @@ class Chatbot:
         similar_questions, similar_contexts, distances, indices = self.faiss_search(question_vector)
         context = similar_contexts[0]
         Answer = self.model_pipeline(similar_questions[0], context)
-        # start_index = context.find(Answer)
-        # end_index = start_index + len(Answer)
-        # output = {
-        #     "user_question": message,
-        #     "answer": self.df['Answer'][indices[0][0]],
-        #     "distance": round(distances[0][0], 4),
-        #     "highlight_start": start_index,
-        #     "highlight_end": end_index
-        # }
-        # return output
         return Answer
 
 if __name__ == "__main__":
@@ -155,8 +140,6 @@ if __name__ == "__main__":
     
     def chat_interface(question, history):
         response = bot._chatbot.predict(question)
-        # highlighted_answer = highlight_text(response["answer"], response["highlight_start"], response["highlight_end"])
-        # return highlighted_answer
         return response
 
     EXAMPLE = ["หลิน ไห่เฟิง มีชื่อเรียกอีกชื่อว่าอะไร" , "ใครเป็นผู้ตั้งสภาเศรษฐกิจโลกขึ้นในปี พ.ศ. 2514 โดยทุกปีจะมีการประชุมที่ประเทศสวิตเซอร์แลนด์", "โปรดิวเซอร์ของอัลบั้มตลอดกาล ของวงคีรีบูนคือใคร", "สกุลเดิมของหม่อมครูนุ่ม นวรัตน ณ อยุธยา คืออะไร"]