Upload app.py

app.py

@@ -115,9 +115,10 @@ class PredictionTask:
                 data = stock_model.fetch_stock_data(self.symbol, outputsize="compact")
                 print(f"Fetched {len(data)} rows for {self.symbol}")
             except Exception as e:
-                print(f"Data fetch error: {e}")
+                error_msg = str(e)
+                print(f"\n[ERROR] {error_msg}\n")
                 self.status = "failed"
-                self.result = {"error": f"Could not fetch data for {self.symbol}: {e}"}
+                self.result = {"error": error_msg}
                 return
 
             if data is None:
@@ -333,9 +334,10 @@ class PredictionTask:
             print(f"Prediction complete for {self.symbol}")
 
         except Exception as e:
+            error_msg = str(e)
             self.status = "failed"
-            self.result = {"error": str(e)}
-            print(f"Prediction task error: {e}")
+            self.result = {"error": error_msg}
+            print(f"\n[ERROR] {error_msg}\n")
             traceback.print_exc()
 
 
@@ -416,7 +418,8 @@ def prediction_status(task_id):
                 "taskId":   task_id,
                 "status":   task.status,
                 "progress": task.progress,
-                "result":   task.result if task.status == "completed" else None,
+                "result":   task.result if task.status in ["completed", "failed"] else None,
+                "error":    task.result.get("error") if (task.status == "failed" and task.result and isinstance(task.result, dict)) else None
             })
         except Exception as e:
             print(f"Error generating status response: {e}")