Update app.py

app.py

@@ -1,7 +1,6 @@
 # =============================================================================
 # By T.Sakai / G.Huet
 # =============================================================================
-
 import os
 import time
 import io
@@ -930,6 +929,10 @@ def replace_graph_placeholders(report_html: str, context: str, initial_query: st
         
         try:
             visual_html = generate_graph_snippet(instructions, context, initial_query, crumbs)
+            prompt_improve = f"Based on the instructions:\n{instructions}\n\nThe context:\n{context}\n\nThe knowledge crumbs:\n{crumbs}\n\nProvide a refined version to this graph code:\n{visual_html}\n\nThe improved result should be EXACTLY in the same html format (no introduction, no conclusion, no code fences)."
+            visual_html = openai_call(prompt_improve, model="o3-mini", max_tokens=10000)
+            visual_html = visual_html.strip().strip("```").strip()
+
             return f'<!-- Graph {placeholder_num} Start -->\n<div class="graph-container" style="margin:20px 0;">\n{visual_html}\n</div>\n<!-- Graph {placeholder_num} End -->'
         except Exception as e:
             logging.error(f"Graph {placeholder_num} failed: {str(e)}")
@@ -1184,6 +1187,9 @@ def replace_visual_placeholders(report_html: str, context: str, initial_query: s
         
         try:
             visual_html = generate_visual_snippet(instructions, context, initial_query, crumbs)
+            prompt_improve = f"Based on the instructions:\n{instructions}\n\nThe context:\n{context}\n\nThe knowledge crumbs:\n{crumbs}\n\nProvide a refined version to this visual code:\n{visual_html}\n\nThe improved result should be EXACTLY in the same html format (no introduction, no conclusion, no code fences)."
+            visual_html = openai_call(prompt_improve, model="o3-mini", max_tokens=10000)
+            visual_html = visual_html.strip().strip("```").strip()
             # Add error boundary and logging
             return f'<!-- Visual {placeholder_num} Start -->\n{visual_html}\n<!-- Visual {placeholder_num} End -->'
         except Exception as e:
@@ -1802,8 +1808,9 @@ Source:source_name [y]
 [[Visual Placeholder n:
 - Purpose of this visual is:...
 - Relevant content to generate it:
-o ...
-o ...
+o ex: arguments
+o ex: lists of elements
+o ex: data points
 o ...
 - Message to convey: ...
 ]]
@@ -1840,6 +1847,7 @@ with:
 - n as the reference number,
 - source_name as the  full name of the main source used and
 - y as the source reference in the reference table.
+- the table containing all the required data has to include data points FROM the learnings / results from the search below
 Important note for graph placeholders:
 - on the line before [[...]] mention the source with the reference number [x] in the form: ""Source: abc [n]" - only one source should be mentioned
 - use p tag for the source and source reference number