Update app.py

app.py

@@ -203,7 +203,7 @@ def view_database():
 
 # Function to download database or HTML file
 def download_file(choice):
-    directory = "/mnt/data"
+    directory = tempfile.gettempdir()
     
     # Ensure the directory exists
     if not os.path.exists(directory):
@@ -238,6 +238,8 @@ def download_file(choice):
 
         # Return the path to the HTML file
         return html_file_path
+    else:
+        raise ValueError("Invalid choice. Please select a valid format.")
         
 # Initialize the database
 init_db()
@@ -265,6 +267,8 @@ def download_interface(choice):
             return file.read(), file_path.split('/')[-1]
     except FileNotFoundError as e:
         return str(e), None
+    except ValueError as e:
+        return str(e), None
 
 # Gradio Blocks
 with gr.Blocks() as demo: