remove ipython functions

src/few_shot_funcs.py

@@ -5,7 +5,6 @@ import inflect
 import pandas as pd
 from typing import Dict
 from datasets import load_dataset
-from IPython.display import display, HTML
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.vectorstores.utils import DistanceStrategy
@@ -52,19 +51,6 @@ class Conversation:
         
         message = {"role": role, "content": content}
         self.messages.append(message)
-            
-    def display_conversation(self) -> None:
-        SEP = '\n'
-        for message in self.messages:
-            if message['role'] == 'system':
-                display(HTML(f'<b>{message["content"]}</b>'))
-            elif message['role'] == 'user':
-                msg_align = message["content"].replace("Title:", "<br><b>Title:</b>")\
-                    .replace("Technical details:", "<br><b>Technical details:</b>").replace("### Feature-bullets:", "<br><b>Feature-bullets:</b>")
-                display(HTML(f'<p style="background-color:White; color:Black; padding:5px;">{msg_align}</p>'))
-            else:
-                msg_align = message["content"].lstrip(SEP).replace(SEP, "<br><br>")
-                display(HTML(f'<p style="background-color:LightGray; color:Black; padding:5px;">{msg_align}</p>'))
 
 def api_call(messages: Dict[str, str], temperature: float = 0.7, top_p: int = 1, n_responses: int = 1) -> dict:
     """