Spaces:
Runtime error
Runtime error
Update utils/_init_.py
Browse files- 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
|