Update main.py

main.py

@@ -278,6 +278,8 @@ def submit_feedback():
 # Content
 # -----------------------
 # ---------- Story Generation Endpoint ----------
+# Configure logging to write to a file
+logging.basicConfig(filename=LOG_FILE_PATH, level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s")
 @app.route('/api/story/generate', methods=['POST'])
 def generate_story_endpoint():
     try:
@@ -407,11 +409,15 @@ def generate_story_endpoint():
             if input_type == "dataframe" and df is not None:
                 try:
                     chart_str = generateResponse(img_prompt, df)
+                    logging.info(f"chart string: {chart_str}")
                     if chart_str and chart_str.startswith("data:image/png;base64,"):
                         base64_data = chart_str.split(",", 1)[1]
-                        chart_bytes = base64.b64decode(base64_data)
+                        logging.info(f"base64 data: {base64_data}")
+                        chart_bytes = base64.b64decode(chart_bytes)
                         image_obj = Image.open(io.BytesIO(chart_bytes))
+                        logging.info(f"Image: {image_obj})
                 except Exception as e:
+                    logging.error(f"error {e}, DataFrame chart generation error")
                     print("DataFrame chart generation error:", e)
 
             # Fallback to generate_image_with_retry