Update main.py

main.py

@@ -64,6 +64,7 @@ async def get_image_for_text(email,query,file: UploadFile = File(...)):
         try:
             sdf = SmartDataframe(df, config={"llm": llm})
             sdf.chat(query)
+            print(df.head())
             image_path = "exports/charts/temp_chart.png"  # Replace with your image's path
             base64str = convert_image_to_base64(image_path)
             return {"id":str(uuid1),"image":base64str}