Update app.py

app.py

@@ -16,7 +16,9 @@ client = InferenceClient(
 history = []
 hist_out= []
 summary =[]
+main_point=[]
 summary.append("")
+main_point.append("")
 def format_prompt(message, history):
   prompt = "<s>"
   for user_prompt, bot_response in history:
@@ -38,7 +40,8 @@ repetition_penalty=1.0,
 def compress_history(formatted_prompt):
     
     seed = random.randint(1,1111111111111111)
-    agent=prompts.COMPRESS_HISTORY_PROMPT.format(history=summary[0])
+    agent=prompts.COMPRESS_HISTORY_PROMPT.format(history=summary[0],focus=main_point[0])
+    
     system_prompt=agent
     temperature = 0.9
     if temperature < 1e-2:
@@ -62,13 +65,14 @@ def compress_history(formatted_prompt):
         output += response.token.text
     #history.append((output,history))
     print(output)
+    print(main_point[0])
     return output
 
 
 def question_generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
 #def question_generate(prompt, history):
     seed = random.randint(1,1111111111111111)
-    agent=prompts.QUESTION_GENERATOR
+    agent=prompts.QUESTION_GENERATOR.format(focus=main_point[0])
     system_prompt=agent
     temperature = float(temperature)
     if temperature < 1e-2:
@@ -105,6 +109,7 @@ def create_valid_filename(invalid_filename: str) -> str:
     return ''.join(char for char in valid_chars if char in allowed_chars)
     
 def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=1048, top_p=0.95, repetition_penalty=1.0,):
+    main_point[0]=prompt
     #print(datetime.datetime.now())
     uid=uuid.uuid4()
     current_time = str(datetime.datetime.now())