Update app.py

app.py

@@ -15,6 +15,7 @@ openai_tools = [{"type": "web_search_preview", "search_context_size": "high", "u
 # JSON parser
 def parse_json_response(response: str) -> list:
     cleaned = response.replace("json", "").replace("```", "").replace("\n", "")
+    print(cleaned)
     return json.loads(cleaned)
 
 
@@ -60,11 +61,11 @@ def build_people_prompt(area: str) -> str:
         Надай список 5 успішних людей (ім`я та прізвище). 
         Шукай все що напряму пов'язано з {area} (компанії, дочірні компанії, співробітники, і тд)""" + JSON_INSTRUCTION
 
-def build_books_prompt(name: str, context: str) -> str:
-    return f"""Ти найкращий пошуковий помічник по книгам у світі. Надай мені список книг, які рекомендував {name} з {context}. 
+def build_books_prompt(name: str, query: str) -> str:
+    return f"""Ти найкращий пошуковий помічник по книгам у світі. Надай мені список книг, які рекомендував {name} з {query}. 
     Мене цікавлять тільки назви книг, не пиши нічого іншого. В відповіді повинні бути лише назва книги. 
-    Проаналізуй всі статті, публікації, згадки у соціальних мережах про {name} з {context} і знайди саме книги які ця людина рекомендувала.
-    Ці книги потрібні щоб віповісти на запит користувача {query} + {context}
+    Проаналізуй всі статті, публікації, згадки у соціальних мережах про {name} з {query} і знайди саме книги які ця людина рекомендувала.
+    Ці книги потрібні щоб віповісти на запит користувача {query}
     Якщо ти нічого не зміг знайти поверни "". БІЛЬШЕ НІЧОГО НЕ ПИШИ."""
     
 # Generator function for progressive output