update app.py

app.py

@@ -12,7 +12,7 @@ from langchain.retrievers.document_compressors import LLMChainExtractor
 from langchain.embeddings import HuggingFaceBgeEmbeddings 
 from langchain.llms import HuggingFacePipeline
 from langchain.vectorstores import Chroma
-
+from templates import all_templates
 
 
 @st.cache_resource(show_spinner=False)
@@ -258,49 +258,10 @@ if __name__=="__main__":
     bar.progress(0.5, "Loading Model.  Please wait.")
     model, tokenizer = load_model(model_name)
     bar.progress(0.9, "Ready to ask? Go ahead and type your question.")
-    time.sleep(2) 
-    bar.empty() 
-
-    all_templates = { "llama_prompt_template" : """<s>[INST]\n<<SYS>>\nYou are a stoic teacher that provide guidance and advice inspired by Stoic philosophy on navigating life's challenges with resilience and inner peace. Emphasize the importance of focusing on what is within one's control and accepting what is not. Encourage the cultivation of virtue, mindfulness, and self-awareness as tools for achieving eudaimonia. Advocate for enduring hardships with fortitude and maintaining emotional balance in all situations. Your response should reflect Stoic principles of living in accordance with nature and embracing the rational order of the universe.
-            You should guide the reader towards a fulfilling life focused on virtue rather than external things because living in accordance with virtue leads to eudaimonia or flourishing.\n\n
-            Give a precise answer to the question based on the context. Don't be verbose.\n\n
-            context:
-            {context}\n<</SYS>>\n\n
-            question:
-            {user_query}
-            [/INST]""",
-
-            "llama_rag_template" :"""<s>[INST]\n<<SYS>>\nGiven the following question and context, summarize the parts that are relevant to answer the question. If none of the context is relevant return NO_OUTPUT.\n\n>
-            - Do not mention quotes.\n\n
-            - Reply using a single sentence.\n\n
-            > Context:\n
-            >>>\n{context}\n>>>\n<</SYS>>\n\n
-            Question: {question}\n
-            [/INST]
-            The relevant parts of the context are:
-            """,
-
-            "prompt_template":"""You are a stoic teacher that provide guidance and advice inspired by Stoic philosophy on navigating life's challenges with resilience and inner peace. Emphasize the importance of focusing on what is within one's control and accepting what is not. Encourage the cultivation of virtue, mindfulness, and self-awareness as tools for achieving eudaimonia. Advocate for enduring hardships with fortitude and maintaining emotional balance in all situations. Your response should reflect Stoic principles of living in accordance with nature and embracing the rational order of the universe.
-                You should guide the reader towards a fulfilling life focused on virtue rather than external things because living in accordance with virtue leads to eudaimonia or flourishing.
-                context:
-                {context}
-
-                question:
-                {user_query}
-
-                Answer:
-                """,
-            "rag_prompt" : """Given the following question and context, summarize the parts that are relevant to answer the question. If none of the context is relevant return NO_OUTPUT.\n\n>
-                - Do not mention quotes.\n\n>
-                - Reply using a single sentence.\n\n>
-
-                Question: {question}\n> Context:\n>>>\n{context}\n>>>\nRelevant parts"""}
-
-    
-
+    #time.sleep(2) 
+    #bar.empty() 
   
     # streamlit chat
-
     user_question = st.chat_input('What do you want to ask ..')
     
     if user_question is not None and user_question!="":