Update app.py

app.py

@@ -204,24 +204,36 @@ async def process_file(file: UploadFile = File(...)):
         else:
             raise HTTPException(status_code=400, detail="Unsupported file type")
 
+        # Convert timestamps to strings
+        for col in df.select_dtypes(include=['datetime64[ns]']).columns:
+            df[col] = df[col].astype(str)
+
         df = df.replace({np.nan: None})
         text_reps = [" ".join(f"{col}: {str(val)}" for col, val in row.items() if val is not None)
                     for _, row in df.iterrows()]
         
         embeddings = model.encode(text_reps)
         
+        # Convert DataFrame to JSON-serializable format
+        raw_data = df.to_dict('records')
+        # Convert any non-serializable types to strings
+        for record in raw_data:
+            for key, value in record.items():
+                if not isinstance(value, (str, int, float, bool, type(None))):
+                    record[key] = str(value)
+        
         return JSONResponse({
             'embeddings': embeddings.tolist(),
             'metadata': {
                 'columns': list(df.columns),
                 'row_count': len(df)
             },
-            'raw_data': df.to_dict('records')
+            'raw_data': raw_data
         })
         
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-
+        
 @app.post("/api/query")
 async def query_data(request: QueryRequest):
     try: