Update main.py

main.py

@@ -90,6 +90,8 @@ class QueryModel(BaseModel):
                                enum=["chat", "report", "report_table"])
     data_format: str = Query(default="Structured data", description="Type of data to extract from the internet",
                              enum=["No presets", "Structured data", "Quantitative data"])
+    generate_charts: bool = Query(default=False, description="Include generated charts")
+    
 
 @cache(expire=604800)
 async def generate_report(query: QueryModel):
@@ -102,7 +104,8 @@ async def generate_report(query: QueryModel):
     data_format = query.data_format
     optimized_search_query = ""
     all_text_with_urls = [("", "")]
-
+    generate_charts = query.generate_charts
+    
     if query.internet:
         search_query = re.sub(r'[^\w\s]', '', description).strip()
         try:
@@ -123,7 +126,7 @@ async def generate_report(query: QueryModel):
     html_report = md_to_html(md_report)
 
     # Render Charts
-    if has_tables(html_report):
+    if generate_charts and has_tables(html_report):
         print("tables found, creating charts")
         try: