Update main.py

main.py

@@ -27,25 +27,13 @@ async def testv1(request: Request, background_tasks: BackgroundTasks):
         body_data = await request.json()
 
         print(body_data)
-        # Create initial batch job record
-        save_data = {
-            'batch_job_id': f"batch_{datetime.utcnow().strftime('%Y%m%d_%H%M%S')}",
-            "batch_job_status": False,
-            "created_at": datetime.utcnow().isoformat()
-        }
-        
-        response = (
-            supabase.table("batch_processing_details")
-            .insert(save_data)
-            .execute()
-        )
+
         
         # Add processing to background tasks
-        background_tasks.add_task(process_batch_job, body_data, save_data['batch_job_id'])
+        background_tasks.add_task(process_batch_job, body_data)
 
         return {'data': 'Batch job is scheduled!'}
         
-        # return {'data': 'Batch job is scheduled!', 'batch_job_id': save_data['batch_job_id']}
            
         
     except Exception as e:
@@ -124,12 +112,12 @@ def adjust_prompt_tokens_v1(prompt: str) -> str:
     return trimmed_text
 
 
-async def process_batch_job(dataset: Dict[str, Any], batch_job_id: str):
+async def process_batch_job(dataset: Dict[str, Any]):
     """
     Background task to process the batch job
     """
     try:
-        logger.info(f"Starting batch processing for job {batch_job_id}")
+
         
         openai_tasks = []
         for ds in dataset.get('data'):
@@ -179,12 +167,11 @@ async def process_batch_job(dataset: Dict[str, Any], batch_job_id: str):
         
         # Update status in Supabase
         supabase.table("batch_processing_details").insert({
-            "batch_job_status": True,
+            "batch_job_status": False,
             "batch_job_id" : batch_job.id
         }).execute()
         
-        logger.info(f"Batch job {batch_job_id} processed successfully")
-        
+        print("Batch sent for processing")        
     except Exception as e:
         logger.error(f"Error processing batch job {batch_job_id}: {str(e)}")
         # Update status with error