Update app.py

app.py

@@ -4,8 +4,8 @@ import time
 import os
 from dotenv import load_dotenv
 from sentence_transformers import SentenceTransformer
-from langchain_community.vectorstores import Chroma
-from langchain_community.embeddings import HuggingFaceEmbeddings 
+#from langchain_community.vectorstores import Chroma
+#from langchain_community.embeddings import HuggingFaceEmbeddings 
 load_dotenv()
 
 CHAT_BOTS = {"Mixtral 8x7B v0.1" :"mistralai/Mixtral-8x7B-Instruct-v0.1"}
@@ -98,6 +98,28 @@ def chat_box() :
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
 
+def formattaPrompt(prompt, systemRole, systemStyle, instruction):
+  input_text = f'''
+  {{
+      "input": {{
+          "role": "system",
+          "content": "{systemRole}", 
+          "style": "{systemStyle}"
+      }},
+      "messages": [
+          {{
+              "role": "instructions",
+              "content": "{instruction} "("{systemStyle}")"
+          }},
+          {{
+              "role": "user",
+              "content": "{input}"
+          }}
+      ]
+  }}
+  '''
+  return input_text
+
 def generate_chat_stream(prompt) :
     links = []
     if st.session_state.rag_enabled :