blank image issue on multiple req

controller.py

@@ -309,81 +309,81 @@ def handle_out_of_range_float(value):
 
 # CHART CODING STARTS FROM HERE
 
-# instructions = """
+instructions = """
 
-# - Please ensure that each value is clearly visible, You may need to adjust the font size, rotate the labels, or use truncation to improve readability (if needed).
-# - For multiple charts, arrange them in a grid format (2x2, 3x3, etc.)
-# - Use colorblind-friendly palette
-# - Read above instructions and follow them.
+- Please ensure that each value is clearly visible, You may need to adjust the font size, rotate the labels, or use truncation to improve readability (if needed).
+- For multiple charts, arrange them in a grid format (2x2, 3x3, etc.)
+- Use colorblind-friendly palette
+- Read above instructions and follow them.
 
-# """
+"""
 
-# # Thread-safe configuration for chart endpoints
-# current_groq_chart_key_index = 0
-# current_groq_chart_lock = threading.Lock()
+# Thread-safe configuration for chart endpoints
+current_groq_chart_key_index = 0
+current_groq_chart_lock = threading.Lock()
 
 # current_langchain_chart_key_index = 0
 # current_langchain_chart_lock = threading.Lock()
 
-# def model():
-#     global current_groq_chart_key_index, current_groq_chart_lock
-#     with current_groq_chart_lock:
-#         if current_groq_chart_key_index >= len(groq_api_keys):
-#             raise Exception("All API keys exhausted for chart generation")
-#         api_key = groq_api_keys[current_groq_chart_key_index]
-#     return ChatGroq(model=model_name, api_key=api_key)
+def model():
+    global current_groq_chart_key_index, current_groq_chart_lock
+    with current_groq_chart_lock:
+        if current_groq_chart_key_index >= len(groq_api_keys):
+            raise Exception("All API keys exhausted for chart generation")
+        api_key = groq_api_keys[current_groq_chart_key_index]
+    return ChatGroq(model=model_name, api_key=api_key)
 
-# def groq_chart(csv_url: str, question: str):
-#     global current_groq_chart_key_index, current_groq_chart_lock
+def groq_chart(csv_url: str, question: str):
+    global current_groq_chart_key_index, current_groq_chart_lock
     
-#     for attempt in range(len(groq_api_keys)):
-#         try:
-#             # Clean cache before processing
-#             cache_db_path = "/workspace/cache/cache_db_0.11.db"
-#             if os.path.exists(cache_db_path):
-#                 try:
-#                     os.remove(cache_db_path)
-#                 except Exception as e:
-#                     print(f"Cache cleanup error: {e}")
+    for attempt in range(len(groq_api_keys)):
+        try:
+            # Clean cache before processing
+            cache_db_path = "/workspace/cache/cache_db_0.11.db"
+            if os.path.exists(cache_db_path):
+                try:
+                    os.remove(cache_db_path)
+                except Exception as e:
+                    print(f"Cache cleanup error: {e}")
 
-#             data = clean_data(csv_url)
-#             with current_groq_chart_lock:
-#                 current_api_key = groq_api_keys[current_groq_chart_key_index]
+            data = clean_data(csv_url)
+            with current_groq_chart_lock:
+                current_api_key = groq_api_keys[current_groq_chart_key_index]
             
-#             llm = ChatGroq(model=model_name, api_key=current_api_key)
+            llm = ChatGroq(model=model_name, api_key=current_api_key)
             
-#             # Generate unique filename using UUID
-#             chart_filename = f"chart_{uuid.uuid4()}.png"
-#             chart_path = os.path.join("generated_charts", chart_filename)
+            # Generate unique filename using UUID
+            chart_filename = f"chart_{uuid.uuid4()}.png"
+            chart_path = os.path.join("generated_charts", chart_filename)
             
-#             # Configure SmartDataframe with chart settings
-#             df = SmartDataframe(
-#                 data,
-#                 config={
-#                     'llm': llm,
-#                     'save_charts': True,  # Enable chart saving
-#                     'open_charts': False,
-#                     'save_charts_path': os.path.dirname(chart_path),  # Directory to save
-#                     'custom_chart_filename': chart_filename  # Unique filename
-#                 }
-#             )
+            # Configure SmartDataframe with chart settings
+            df = SmartDataframe(
+                data,
+                config={
+                    'llm': llm,
+                    'save_charts': True,  # Enable chart saving
+                    'open_charts': False,
+                    'save_charts_path': os.path.dirname(chart_path),  # Directory to save
+                    'custom_chart_filename': chart_filename  # Unique filename
+                }
+            )
             
-#             answer = df.chat(question + instructions)
+            answer = df.chat(question + instructions)
             
-#             if process_answer(answer):
-#                 return "Chart not generated"
-#             return answer
+            if process_answer(answer):
+                return "Chart not generated"
+            return answer
 
-#         except Exception as e:
-#             error = str(e)
-#             if "429" in error:
-#                 with current_groq_chart_lock:
-#                     current_groq_chart_key_index = (current_groq_chart_key_index + 1) % len(groq_api_keys)
-#             else:
-#                 print(f"Chart generation error: {error}")
-#                 return {"error": error}
+        except Exception as e:
+            error = str(e)
+            if "429" in error:
+                with current_groq_chart_lock:
+                    current_groq_chart_key_index = (current_groq_chart_key_index + 1) % len(groq_api_keys)
+            else:
+                print(f"Chart generation error: {error}")
+                return {"error": error}
     
-#     return {"error": "All API keys exhausted for chart generation"}
+    return {"error": "All API keys exhausted for chart generation"}
 
 
 
@@ -572,8 +572,8 @@ def handle_out_of_range_float(value):
 
 
 # Global locks for key rotation (chart endpoints)
-current_groq_chart_key_index = 0
-current_groq_chart_lock = threading.Lock()
+# current_groq_chart_key_index = 0
+# current_groq_chart_lock = threading.Lock()
 current_langchain_chart_key_index = 0
 current_langchain_chart_lock = threading.Lock()
 
@@ -584,83 +584,83 @@ print("Available CPUs:", max_cpus)
 process_executor = ProcessPoolExecutor(max_workers=4)
 
 # --- GROQ-BASED CHART GENERATION ---
-def groq_chart(csv_url: str, question: str):
-    """
-    Generate a chart using the groq-based method.
-    Modifications:
-      • No deletion of a shared cache file (avoid interference).
-      • After chart generation, close all matplotlib figures.
-      • Return the full path of the saved chart.
-    """
-    global current_groq_chart_key_index, current_groq_chart_lock
+# def groq_chart(csv_url: str, question: str):
+#     """
+#     Generate a chart using the groq-based method.
+#     Modifications:
+#       • No deletion of a shared cache file (avoid interference).
+#       • After chart generation, close all matplotlib figures.
+#       • Return the full path of the saved chart.
+#     """
+#     global current_groq_chart_key_index, current_groq_chart_lock
     
-    for attempt in range(len(groq_api_keys)):
-        try:
-            # Instead of deleting a global cache file, you might later configure a per-request cache.
-            cache_db_path = "/app/cache/cache_db_0.11.db"
-            if os.path.exists(cache_db_path):
-                try:
-                    os.remove(cache_db_path)
-                    print(f"Deleted cache DB file: {cache_db_path}")
-                except Exception as e:
-                    print(f"Error deleting cache DB file: {e}")
+#     for attempt in range(len(groq_api_keys)):
+#         try:
+#             # Instead of deleting a global cache file, you might later configure a per-request cache.
+#             cache_db_path = "/app/cache/cache_db_0.11.db"
+#             if os.path.exists(cache_db_path):
+#                 try:
+#                     os.remove(cache_db_path)
+#                     print(f"Deleted cache DB file: {cache_db_path}")
+#                 except Exception as e:
+#                     print(f"Error deleting cache DB file: {e}")
                     
-            chart_dir = "generated_charts"
-            if not os.path.exists(chart_dir):
-               os.makedirs(chart_dir)
+#             chart_dir = "generated_charts"
+#             if not os.path.exists(chart_dir):
+#                os.makedirs(chart_dir)
                     
-            data = clean_data(csv_url)
-            with current_groq_chart_lock:
-                current_api_key = groq_api_keys[current_groq_chart_key_index]
+#             data = clean_data(csv_url)
+#             with current_groq_chart_lock:
+#                 current_api_key = groq_api_keys[current_groq_chart_key_index]
             
-            llm = ChatGroq(model=model_name, api_key=current_api_key)
+#             llm = ChatGroq(model=model_name, api_key=current_api_key)
             
-            # Generate a unique filename and full path for the chart
-            chart_filename = f"chart_{uuid.uuid4().hex}.png"
-            chart_path = os.path.join("generated_charts", chart_filename)
+#             # Generate a unique filename and full path for the chart
+#             chart_filename = f"chart_{uuid.uuid4().hex}.png"
+#             chart_path = os.path.join("generated_charts", chart_filename)
             
-            # Configure your dataframe tool (e.g. using SmartDataframe) to save charts.
-            # (Assuming your SmartDataframe uses these settings to save charts.)
-            from pandasai import SmartDataframe  # Import here if not already imported
-            df = SmartDataframe(
-                data,
-                config={
-                    'llm': llm,
-                    'save_charts': True,
-                    'open_charts': False,
-                    'save_charts_path': os.path.dirname(chart_path),
-                    'custom_chart_filename': chart_filename
-                }
-            )
+#             # Configure your dataframe tool (e.g. using SmartDataframe) to save charts.
+#             # (Assuming your SmartDataframe uses these settings to save charts.)
+#             from pandasai import SmartDataframe  # Import here if not already imported
+#             df = SmartDataframe(
+#                 data,
+#                 config={
+#                     'llm': llm,
+#                     'save_charts': True,
+#                     'open_charts': False,
+#                     'save_charts_path': os.path.dirname(chart_path),
+#                     'custom_chart_filename': chart_filename
+#                 }
+#             )
             
-            # Append any extra instructions if needed
-            instructions = """
-            - Ensure each value is clearly visible.
-            - Adjust font sizes, rotate labels if necessary.
-            - Use a colorblind-friendly palette.
-            - Arrange multiple charts in a grid if needed.
-            """
-            answer = df.chat(question + instructions)
+#             # Append any extra instructions if needed
+#             instructions = """
+#             - Ensure each value is clearly visible.
+#             - Adjust font sizes, rotate labels if necessary.
+#             - Use a colorblind-friendly palette.
+#             - Arrange multiple charts in a grid if needed.
+#             """
+#             answer = df.chat(question + instructions)
             
-            # Make sure to close figures so they don't conflict between processes
-            plt.close('all')
+#             # Make sure to close figures so they don't conflict between processes
+#             plt.close('all')
             
-            # If process_answer indicates a problem, return a failure message.
-            if process_answer(answer):
-                return "Chart not generated"
-            # Return the chart path that was used for saving
-            return chart_path
+#             # If process_answer indicates a problem, return a failure message.
+#             if process_answer(answer):
+#                 return "Chart not generated"
+#             # Return the chart path that was used for saving
+#             return chart_path
 
-        except Exception as e:
-            error = str(e)
-            if "429" in error:
-                with current_groq_chart_lock:
-                    current_groq_chart_key_index = (current_groq_chart_key_index + 1) % len(groq_api_keys)
-            else:
-                print(f"Groq chart generation error: {error}")
-                return {"error": error}
+#         except Exception as e:
+#             error = str(e)
+#             if "429" in error:
+#                 with current_groq_chart_lock:
+#                     current_groq_chart_key_index = (current_groq_chart_key_index + 1) % len(groq_api_keys)
+#             else:
+#                 print(f"Groq chart generation error: {error}")
+#                 return {"error": error}
     
-    return {"error": "All API keys exhausted for chart generation"}
+#     return {"error": "All API keys exhausted for chart generation"}
 
 
 # --- LANGCHAIN-BASED CHART GENERATION ---