fix: bugs regarding event update; improve debug experience

app.py

@@ -21,6 +21,9 @@ from pypandoc.pandoc_download import download_pandoc
 import os
 import json
 
+## debug
+from icecream import ic
+
 logger = mylogger(__name__, "%(asctime)s:%(levelname)s:%(message)s")
 info = logger.info
 
@@ -33,7 +36,7 @@ def init():
 ## Config Functions
 
 
-def set_same_cheap_strong(set_same: bool, cheap_base, cheap_key, cheap_model):
+def set_same_cheap_strong(set_same: bool, cheap_base, cheap_key, cheap_model, strong_base, strong_key, strong_model):
     # setup_zone = gr.Accordion("AI setup (OpenAI-compatible LLM API)", open=True)
     if set_same:
         return (
@@ -46,11 +49,11 @@ def set_same_cheap_strong(set_same: bool, cheap_base, cheap_key, cheap_model):
         )
     else:
         return (
-            gr.Textbox(value=cheap_base, label="API Base", interactive=True),
+            gr.Textbox(value=strong_base, label="API Base", interactive=True),
             gr.Textbox(
-                value=cheap_key, label="API key", type="password", interactive=True
+                value=strong_key, label="API key", type="password", interactive=True
             ),
-            gr.Textbox(value=cheap_model, label="Model ID", interactive=True),
+            gr.Textbox(value=strong_model, label="Model ID", interactive=True),
             # setup_zone,
         )
 
@@ -111,6 +114,7 @@ def finalize_letter_txt(api_base, api_key, api_model, debug_CoT):
     for response in taskAI.purify_letter(full_text=debug_CoT):
         result += response.delta
         yield result
+    
 
 
 def finalize_letter_pdf(api_base, api_key, api_model, jd, cv, cover_letter_text, is_debug):
@@ -118,6 +122,8 @@ def finalize_letter_pdf(api_base, api_key, api_model, jd, cv, cover_letter_text,
     taskAI = TaskAI(cheapAPI, temperature=0.1, max_tokens=100)
     meta_data = next(taskAI.get_jobapp_meta(JD=jd, CV=cv))
     pdf_context = json.loads(meta_data)
+    if is_debug:
+        ic(pdf_context)
     pdf_context["letter_body"] = cover_letter_text
     return meta_data, compile_pdf(
         pdf_context,
@@ -205,16 +211,20 @@ with gr.Blocks(
 
     is_same_cheap_strong.change(
         fn=set_same_cheap_strong,
-        inputs=[is_same_cheap_strong, cheap_base, cheap_key, cheap_model],
+        inputs=[is_same_cheap_strong, cheap_base, cheap_key, cheap_model, strong_base, strong_key, strong_model],
         outputs=[strong_base, strong_key, strong_model],
     )
 
     infer_btn.click(
         fn=set_same_cheap_strong,
-        inputs=[is_same_cheap_strong, cheap_base, cheap_key, cheap_model],
+        inputs=[is_same_cheap_strong, cheap_base, cheap_key, cheap_model, strong_base, strong_key, strong_model],
         outputs=[strong_base, strong_key, strong_model],
     ).success(
         fn=prepare_input, inputs=[jd_info, cv_file, cv_text], outputs=[jd_info, cv_text]
+    ).success(
+        fn=lambda: gr.Accordion("Reformatting", open=True),
+        inputs=None,
+        outputs=[reformat_zone],
     ).success(
         fn=run_refine,
         inputs=[cheap_base, cheap_key, cheap_model, jd_info, cv_text],

requirements.txt

@@ -9,5 +9,6 @@ llama-index-llms-openai-like
 tiktoken
 # lib
 requests
+icecream
 # dev tools
 ruff

taskAI.py

@@ -7,6 +7,8 @@ from llama_index.core import ChatPromptTemplate
 from util import mylogger
 from util import checkAPI
 
+from icecream import ic
+
 logger = mylogger(__name__, "%(asctime)s:%(filename)s:%(levelname)s:%(message)s")
 ## define templates
 
@@ -129,7 +131,7 @@ class TaskAI(OpenAILike):
         meta_CV = self.chat(
             JSON_API.format_messages(
                 template=keys_to_template(
-                    ["applicantFullNname", "applicantContactInformation"]
+                    ["applicantFullName", "applicantContactInformation"]
                 ),
                 content=CV,
             )
@@ -139,7 +141,7 @@ class TaskAI(OpenAILike):
             meta_JD = json.loads(meta_JD.strip())
             meta_CV = json.loads(meta_CV.strip())
         except Exception as e:
-            print(e)
+            ic(e)
             raise ValueError(
                 f"AI didn't return a valid JSON string. Try again or consider a better model for CheapAI. \n{meta_JD}\n{meta_CV}"
             )

taskNonAI.py

@@ -11,6 +11,8 @@ from string import Template
 from datetime import datetime
 from pathlib import Path
 
+from icecream import ic
+
 
 def file_to_html(file_path: str) -> str:
     return pypandoc.convert_file(file_path, "html")
@@ -56,7 +58,6 @@ def _ensure_no_signature_in_body(cover_letter_body: str) -> str:
     if not cover_letter_body.strip().endswith(","):
         # remove last line
         cover_letter_body = "\n".join(cover_letter_body.split("\n")[:-1])
-        print(cover_letter_body)
     return cover_letter_body
 
 def compile_pdf(

util.py

@@ -4,6 +4,8 @@ from urllib.parse import urlparse
 import requests
 import logging
 
+from icecream import ic
+
 from typing import Generator
 
 
@@ -38,7 +40,8 @@ def is_valid_openai_api_key(api_base: str, api_key: str) -> bool:
     headers = {"Authorization": f"Bearer {api_key}"}
     test_url = f"{api_base}/models"
     response = requests.get(test_url, headers=headers)
-    print(response.json())
+    if response.status_code != 200:
+        ic(response.json())
     return response.status_code == 200