max length from 10 --> 8

python_code_executor_service.py

@@ -27,28 +27,22 @@ class CodeResponse(BaseModel):
     language: str = "python"
     code: str
 
-
 class ChartSpecification(BaseModel):
     """Details about requested charts"""
     image_description: str
     code: Optional[str] = None
 
-
 class AnalysisOperation(BaseModel):
     """Container for a single analysis operation with its code and result"""
     code: CodeResponse
     result_var: str
 
-
 class CsvChatResult(BaseModel):
     """Structured response for CSV-related AI interactions"""
     casual_response: str
-    # analysis_operations: Optional[AnalysisOperation] 
-    # charts: Optional[ChartSpecification] 
     analysis_operations: Optional[List[AnalysisOperation]]
     charts: Optional[List[ChartSpecification]]
 
-
 class PythonExecutor:
     """Handles execution of Python code with comprehensive data analysis libraries"""
     
@@ -196,120 +190,68 @@ class PythonExecutor:
         return str(result)
     
     async def process_response(self, response: CsvChatResult, chat_id: str) -> str:
-     """Process the response with proper variable handling and error checking"""
-     output_parts = [response.casual_response]
+        """Process the response with proper variable handling and error checking"""
+        output_parts = [response.casual_response]
 
-     # Process analysis operations if they exist
-     if response.analysis_operations is not None:
-        for operation in response.analysis_operations:
-            try:
-                if operation and operation.code and operation.code.code:
-                    execution_result = self.execute_code(operation.code.code)
+        # Process analysis operations if they exist
+        if response.analysis_operations is not None:
+            for operation in response.analysis_operations:
+                try:
+                    if operation and operation.code and operation.code.code:
+                        execution_result = self.execute_code(operation.code.code)
                 
-                    # Get the result from locals
-                    result = self.exec_locals.get(operation.result_var)
+                        # Get the result from locals
+                        result = self.exec_locals.get(operation.result_var)
                 
-                    if execution_result.get('error'):
-                        output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
-                        output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
-                    elif result is not None:
-                        # Handle empty/None results
-                        if result is None or (hasattr(result, '__len__') and len(result) == 0):
-                            output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
+                        if execution_result.get('error'):
+                            output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
+                            output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
+                        elif result is not None:
+                            # Handle empty/None results
+                            if result is None or (hasattr(result, '__len__') and len(result) == 0):
+                                output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
+                            else:
+                                output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
+                                output_parts.append("```python\n" + self._format_result(result) + "\n```")
                         else:
-                            output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
-                            output_parts.append("```python\n" + self._format_result(result) + "\n```")
+                            output_str = execution_result.get('output', '').strip()
+                            if output_str:
+                                output_parts.append("```\n" + output_str + "\n```")
                     else:
-                        output_str = execution_result.get('output', '').strip()
-                        if output_str:
-                            output_parts.append("```\n" + output_str + "\n```")
-                else:
-                    output_parts.append("\n⚠️ Invalid analysis operation - missing code or result variable")
-            except Exception as e:
-                output_parts.append(f"\n❌ Error processing analysis operation: {str(e)}")
-                if operation:
-                    output_parts.append(f"Operation: {operation.result_var}")
+                        output_parts.append("\n⚠️ Invalid analysis operation - missing code or result variable")
+                except Exception as e:
+                    output_parts.append(f"\n❌ Error processing analysis operation: {str(e)}")
+                    if operation:
+                        output_parts.append(f"Operation: {operation.result_var}")
 
         # Process charts if they exist
         if response.charts is not None:
-         for chart in response.charts:
-            try:
-                if chart and (chart.code or chart.image_description):
-                    if chart.code:
-                        chart_result = self.execute_code(chart.code)
-                        if chart_result.get('plots'):
-                            for plot_data in chart_result['plots']:
-                                try:
-                                    public_url = await self.save_plot_to_supabase(
-                                        plot_data=plot_data,
-                                        description=chart.image_description,
-                                        chat_id=chat_id
-                                    )
-                                    output_parts.append(f"\n🖼️ {chart.image_description}")
-                                    output_parts.append(f"![{chart.image_description}]({public_url})")
-                                except Exception as e:
-                                    output_parts.append(f"\n⚠️ Error uploading chart: {str(e)}")
-                        elif chart_result.get('error'):
-                            output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
+            for chart in response.charts:
+                try:
+                    if chart and (chart.code or chart.image_description):
+                        if chart.code:
+                            chart_result = self.execute_code(chart.code)
+                            if chart_result.get('plots'):
+                                for plot_data in chart_result['plots']:
+                                    try:
+                                        public_url = await self.save_plot_to_supabase(
+                                            plot_data=plot_data,
+                                            description=chart.image_description,
+                                            chat_id=chat_id
+                                        )
+                                        output_parts.append(f"\n🖼️ {chart.image_description}")
+                                        output_parts.append(f"![{chart.image_description}]({public_url})")
+                                    except Exception as e:
+                                        output_parts.append(f"\n⚠️ Error uploading chart: {str(e)}")
+                            elif chart_result.get('error'):
+                                output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
+                            else:
+                                output_parts.append(f"\n⚠️ No chart generated for '{chart.image_description}'")
                         else:
-                            output_parts.append(f"\n⚠️ No chart generated for '{chart.image_description}'")
+                            output_parts.append(f"\n⚠️ No code provided for chart: {chart.image_description}")
                     else:
-                        output_parts.append(f"\n⚠️ No code provided for chart: {chart.image_description}")
-                else:
-                    output_parts.append("\n⚠️ Invalid chart specification")
-            except Exception as e:
-                output_parts.append(f"\n❌ Error processing chart: {str(e)}")
+                        output_parts.append("\n⚠️ Invalid chart specification")
+                except Exception as e:
+                    output_parts.append(f"\n❌ Error processing chart: {str(e)}")
 
-        return "\n".join(output_parts)
-    
-    # async def process_response(self, response: CsvChatResult, chat_id: str) -> str:
-    #  """Process the response with proper variable handling"""
-    #  output_parts = [response.casual_response]
-    
-    #  # Process analysis operations first
-    #  for operation in response.analysis_operations:
-    #      execution_result = self.execute_code(operation.code.code)
-        
-    #      # Get the result from locals
-    #      result = self.exec_locals.get(operation.result_var)
-        
-    #      if execution_result['error']:
-    #          output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
-    #          output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
-    #      elif result is not None:
-    #          # Handle empty/None results
-    #          if result is None or (hasattr(result, '__len__') and len(result) == 0):
-    #              output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
-    #          else:
-    #              output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
-    #              output_parts.append("```python\n" + self._format_result(result) + "\n```")
-    #      else:
-    #          output_str = execution_result['output'].strip()
-    #          if output_str:
-    #              output_parts.append("```\n" + output_str + "\n```")
-                
-    
-    #  # Process charts after all operations
-    #  if response.charts:
-    #      output_parts.append("\n📊 Visualizations:")
-    #      for chart in response.charts:
-    #          if chart.code:
-    #              chart_result = self.execute_code(chart.code)
-    #              if chart_result['plots']:
-    #                  for plot_data in chart_result['plots']:
-    #                      try:
-    #                          public_url = await self.save_plot_to_supabase(
-    #                              plot_data=plot_data,
-    #                              description=chart.image_description,
-    #                              chat_id=chat_id
-    #                          )
-    #                          output_parts.append(f"\n🖼️ {chart.image_description}")
-    #                          output_parts.append(f"![{chart.image_description}]({public_url})")
-    #                      except Exception as e:
-    #                         output_parts.append(f"\n⚠️ Values are missing - Error uploading chart: {str(e)}")
-    #              elif chart_result['error']:
-    #                  output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
-    #              else:
-    #                  output_parts.append(f"\n⚠️ Values are missing - No chart generated for '{chart.image_description}'")
-    
-    #  return "\n".join(output_parts)
+        return "\n".join(output_parts)