Update app.py

app.py

@@ -203,24 +203,21 @@ def view_database():
 
 # Function to download database or image
 def download_file(choice):
-    # Ensure the directory exists
     directory = "/mnt/data"
     if not os.path.exists(directory):
         os.makedirs(directory)
 
     if choice == "Database (.db)":
-        # Provide the path to the database file
         return 'results.db'
     elif choice == "Database (.html)":
         conn = sqlite3.connect('results.db')
         df = pd.read_sql_query("SELECT * FROM results", conn)
         conn.close()
 
-        # Save the DataFrame as an HTML file
         html_file_path = os.path.join(directory, "results.html")
         df.to_html(html_file_path, index=False)
 
-        return html_file_path
+        return html_file_path  # Ensure the path is returned as a string
 
 # Initialize the database
 init_db()
@@ -228,7 +225,7 @@ init_db()
 # Gradio Interface
 def interface(name, age, medical_record, sex, input_image):
     if input_image is None:
-        return "Please upload an image."
+        return None, "Please upload an image.", None
 
     output_image, raw_result = predict_image(input_image, name, age, medical_record, sex)
     submit_status = submit_result(name, age, medical_record, sex, input_image, output_image, raw_result)