Update app.py

app.py

@@ -1,29 +1,56 @@
 import pandas as pd
 import joblib
 import gradio as gr
-from datetime import datetime
-
+from datetime import datetime, timedelta
 
+# Load model
 model = joblib.load('random_forest_model.pkl')
 
-
-def predict_task_duration(duration, deadline):
-    start_date = datetime(2024, 10, 20)
-    deadline_days = (pd.to_datetime(deadline) - start_date).days
-    input_data = pd.DataFrame([[duration, deadline_days]], columns=['duration', 'deadline_days'])
+def predict_task_duration(task_name, duration, deadline):
+    try:
+        start_date = datetime(2024, 10, 20)
+        deadline_date = datetime.strptime(deadline, '%Y-%m-%d')
+        deadline_days = (deadline_date - start_date).days
+        
+        input_data = pd.DataFrame({
+            'duration': [duration],
+            'deadline_days': [deadline_days]
+        })
+        
+        priority = model.predict(input_data)
+        priority_map = {
+            1: "Rendah",
+            2: "Sedang",
+            3: "Tinggi"
+        }
+        
+        result = priority_map.get(priority[0], "Tidak diketahui")
+        return f'Prioritas Tugas: {result}'
     
+    except Exception as e:
+        return f"Error: {str(e)}"
 
-    priority = model.predict(input_data)
-    return f'Prioritas Tugas: {priority[0]}'
-
-
+# Membuat interface Gradio
 iface = gr.Interface(
     fn=predict_task_duration,
     inputs=[
-        gr.Dropdown(choices=["Meeting", "Bekerja", "Belajar"], label="Nama Tugas"),
-        gr.Slider(minimum=1, maximum=10, value=5, label="Durasi Tugas (dalam jam)")
+        gr.Dropdown(
+            choices=["Meeting", "Bekerja", "Belajar"],
+            label="Nama Tugas"
+        ),
+        gr.Slider(
+            minimum=1,
+            maximum=10,
+            value=5,
+            step=0.1,
+            label="Durasi Tugas (dalam jam)"
+        ),
+        gr.Date(label="Deadline")  # Menambahkan input deadline
     ],
-    outputs="text"
+    outputs=gr.Text(label="Hasil Prediksi"),
+    title="Sistem Prediksi Prioritas Tugas",
+    description="Masukkan nama tugas, durasi, dan deadline untuk memprediksi prioritasnya"
 )
 
+# Launch the interface
 iface.launch()