Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from pathlib import Path
 from datetime import datetime
-import os
 
 # Application title
 st.set_page_config(page_title="To-Do List \n Notebook", page_icon="📝", layout="centered")
@@ -45,9 +44,14 @@ def save_uploaded_file(uploaded_file):
         st.error(f"Error saving file: {e}")
         return None
 
-# Function to add a new task
+# Function to add a new task with exception handling
 def add_task(title, description, image_files, audio_file, bg_color):
     try:
+        if not title.strip():
+            raise ValueError("Title cannot be empty.")
+        if not description.strip():
+            raise ValueError("Description cannot be empty.")
+        
         new_task = {
             "title": title,
             "description": description,
@@ -60,20 +64,15 @@ def add_task(title, description, image_files, audio_file, bg_color):
 
         # Clear the inputs after adding the task
         clear_fields()  # Call the function to clear fields
+    except ValueError as ve:
+        st.error(f"Error: {ve}")
     except Exception as e:
         st.error(f"Error adding task: {e}")
 
 # Function to delete a task
 def delete_task(index):
     try:
-        task = st.session_state.tasks.pop(index)
-        # Delete associated files from the 'uploads' folder
-        for file_path in task["images"]:
-            if os.path.exists(file_path):
-                os.remove(file_path)
-        if task["audio"] and os.path.exists(task["audio"]):
-            os.remove(task["audio"])
-        
+        st.session_state.tasks.pop(index)
         st.success("Task deleted successfully!")
         st.session_state.editing_index = None  # Reset editing index after deletion
     except Exception as e:
@@ -95,24 +94,11 @@ def edit_task(index, new_title, new_description, new_images, new_audio, new_bg_c
 
 # Function to clear the fields after a task is added or edited
 def clear_fields():
-    # Clear text inputs
     st.session_state.title = ""
     st.session_state.description = ""
     st.session_state.bg_color = "#ffffff"
-    
-    # Clear file inputs from session state
     st.session_state.image_files = []
     st.session_state.audio_file = None
-    
-    # Delete the uploaded files from the 'uploads' directory
-    for file_path in st.session_state.image_files:
-        if os.path.exists(file_path):
-            os.remove(file_path)
-    
-    if st.session_state.audio_file and os.path.exists(st.session_state.audio_file):
-        os.remove(st.session_state.audio_file)
-    
-    # Reset editing index
     st.session_state.editing_index = None  # Reset editing index
 
 # Add task section