Update app.py

app.py

@@ -177,17 +177,22 @@ def fine_tune_report(adjustment_request: str, openai_api_key: str, serpapi_api_k
     logging.info("fine_tune_report: Starting fine-tuning process based on the adjustment request.")
 
     # Step 1: Get unique plain text string(s) (without HTML) that identify the targeted area.
-    prompt_identify = (
-        f"You are a meticulous technical editor. Below is the full report HTML and a "
-        f"user adjustment request. Extract one or more unique plain-text string(s) (without any HTML tags or formatting) "
-        f"that uniquely appear in the area(s) targeted by the adjustment request. Output them in a JSON object with the key "
-        f"\"identified_unique_strings\" mapped to a list of strings. Ensure these strings exactly match the content in the report.\n\n"
-        f"Full Report HTML:\n{report_html}\n\n"
-        f"User Adjustment Request:\n{adjustment_request}\n\n"
-        f"Only output valid JSON."
+    prompt_identify = (f"""
+You are a meticulous technical editor. Below is the full report HTML and a user adjustment request.
+Extract one or more unique plain-text string(s) (without any HTML tags or formatting) that uniquely appear in the area(s) targeted by the adjustment request.
+Output them in a JSON object with the key "identified_unique_strings" mapped to a list of strings.
+Ensure these strings exactly match the content in the report.
+
+Full Report HTML:
+{report_html}
+
+User Adjustment Request:
+{adjustment_request}
+
+Only output valid JSON."""
     )
     
-    response_identify = openai_call(prompt=prompt_identify, model="o3-mini", max_tokens_param=1500, temperature=0)
+    response_identify = openai_call(prompt=prompt_identify, model="o3-mini", max_tokens_param=5000, temperature=0)
     logging.info(f"fine_tune_report: Raw unique string identification response: {response_identify}")
     
     try:
@@ -1207,25 +1212,6 @@ Note: no need to add a reference table at the end of the focus box, all the refe
     result = result.strip().strip("```").strip()
     logging.info(f"The code produced for this focus placeholder:\n{placeholder_text}\n\n {result}\n\n")
     return result
-
-def replace_focus_placeholders(report_html: str, context: str, initial_query: str, crumbs: str) -> str:
-    pattern = r"\[\[Focus Placeholder (\d+):(.*?)\]\]"  # Capture placeholder number
-    replacements = []
-
-    def placeholder_replacer(match):
-        placeholder_num = match.group(1)
-        instructions = match.group(2).strip()
-        logging.info(f"Generating focus box {placeholder_num}")
-        
-        try:
-            focus_html = generate_focus_snippet(instructions, context, initial_query, crumbs)
-            # Add error boundary and logging
-            return f'<!-- Focus {placeholder_num} Start -->\n{focus_html}\n<!-- Focus {placeholder_num} End -->'
-        except Exception as e:
-            logging.error(f"Focus {placeholder_num} failed: {str(e)}")
-            return f'<!-- ERROR GENERATING VISUAL {placeholder_num} -->'
-
-    return re.sub(pattern, placeholder_replacer, report_html, flags=re.DOTALL)
     
 def get_random_header():
     headers = [