Update app.py

app.py

@@ -12,41 +12,19 @@ cv_json = os.getenv('CV_JSON')
 
 def predict(message, history):
     if find(message):
-      # prompt = (
-      # # f"As Jisu(she/her/hers)'s personal assistant,"
-      # # f"Given that: {question_json}, How can I assist with information on: {message}"
-      # )
-     
-    # prompt = "As Jisu(she/her/hers)'s personal assistant, Given that: {}, How can I assist with information on: {}".format(question_json, message)
         prompt = os.getenv('PREDICT_PROMPT').format(question_json, message)
-          
         response = llm([HumanMessage(content=prompt)])
-          
         return response.content
-      
-    # prompt = (
-    #     # os.getenv('PREDICT_PROMPT1')
-    #     f"As Jisu(she/her/hers)'s personal assistant,"
-    #     f"Given that: {cv_json}, How can I assist with information on: {message}"
-    #     f"If source does not contains relevant information, I will state that the information is not available."
-    # )
 
-    # prompt = "As Jisu(she/her/hers)'s personal assistant, Given that: {}, How can I assist with information on: {}. If the source does not contain relevant information, I will state that the information is not available.".format(cv_json, message)
     prompt = os.getenv('PREDICT_PROMPT1').format(cv_json, message)
 
     response = llm([HumanMessage(content=prompt)])
     return response.content
 
 def find(message):
-    prompt = (
-        "Given the list of questions about Jisu's CV: \n"
-        "- What are Jisu's current projects?\n"
-        "- What are Jisu's publications?\n"
-        "- How can I reach out to Jisu?\n"
-        f"Determine if the following query matches any of the topics above: '{message}'. Answer 'Yes' if it matches, otherwise answer 'No'."
-    )
-    
+    prompt = os.getenv('FIND_PROMPT').format(message)
     response = llm([HumanMessage(content=prompt)])
+    
     if response.content.strip() == 'Yes':
         return True
     else:
@@ -57,7 +35,6 @@ examples = [
     "What are Jisu's current projects?",
     "What are Jisu's publications?",
     "How can I reach out to Jisu?",
-    # "How is the answer generated?"
 ]
 
 with gr.Blocks(theme='gradio/soft', fill_height=True) as demo: