Sivainti commited on
Commit
4e2b778
·
verified ·
1 Parent(s): d100d6e

Update utils/_init_.py

Browse files
Files changed (1) hide show
  1. utils/_init_.py +15 -0
utils/_init_.py CHANGED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ from datetime import timedelta
3
+
4
+ def predict_timeline(df):
5
+ # Dummy logic for testing
6
+ df['Start'] = pd.Timestamp.now()
7
+ df['Finish'] = df['Start'] + pd.to_timedelta(7, unit='D')
8
+ df['Task'] = df.iloc[:, 0] # Assume first column is task name
9
+ df['Risk'] = ['Low'] * len(df)
10
+
11
+ summary = "Timeline generated for {} tasks.".format(len(df))
12
+ risks = [{"text": task, "start": 0, "end": len(task), "entity": risk}
13
+ for task, risk in zip(df['Task'], df['Risk'])]
14
+
15
+ return df[['Task', 'Start', 'Finish', 'Risk']], risks, summary