Update app.py (#3)

- Update app.py (0bd84ca788d4880c3c6df422b2a06e44cac46110)


Co-authored-by: Suri <SuriRaja@users.noreply.huggingface.co>

app.py

@@ -1,52 +1,31 @@
-import sys
+import gradio as gr
+import pandas as pd
 import os
 
-# Add the current directory to the Python path
-sys.path.append(os.path.dirname(os.path.abspath(__file__)))
-
-# Import models
-from models.user_model import User
-from models.appointment_model import Appointment
-from models.prescription_model import Prescription
-from models.report_model import Report
-from models.medical_history_model import MedicalHistory
-
-# Define the directory for CSV files
-DATA_DIR = 'data'
-
-# Define paths to each CSV file
-USER_CSV = os.path.join(DATA_DIR, 'users.csv')
-APPOINTMENT_CSV = os.path.join(DATA_DIR, 'appointments.csv')
-PRESCRIPTION_CSV = os.path.join(DATA_DIR, 'prescriptions.csv')
-REPORT_CSV = os.path.join(DATA_DIR, 'reports.csv')
-MEDICAL_HISTORY_CSV = os.path.join(DATA_DIR, 'medical_history.csv')
-
-def load_all_data():
-    # Load data from each model's CSV file
-    users = User.load_from_csv(USER_CSV)
-    appointments = Appointment.load_from_csv(APPOINTMENT_CSV)
-    prescriptions = Prescription.load_from_csv(PRESCRIPTION_CSV)
-    reports = Report.load_from_csv(REPORT_CSV)
-    medical_history = MedicalHistory.load_from_csv(MEDICAL_HISTORY_CSV)
-
-    # Print out the loaded data for verification
-    print(f"Loaded {len(users)} users.")
-    print(f"Loaded {len(appointments)} appointments.")
-    print(f"Loaded {len(prescriptions)} prescriptions.")
-    print(f"Loaded {len(reports)} reports.")
-    print(f"Loaded {len(medical_history)} medical history records.")
-
-    # Return all the data for further processing
-    return users, appointments, prescriptions, reports, medical_history
-
-def main():
-    print("Initializing MediConsult App...")
-
-    # Load all data from CSV files
-    users, appointments, prescriptions, reports, medical_history = load_all_data()
-
-    # Additional logic to handle the application workflow
-    # For example, managing user authentication, processing appointment requests, etc.
+# Define the upload directory relative to the app's location
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+UPLOAD_DIR = BASE_DIR
+
+def upload_file(file):
+    try:
+        # Save the uploaded file to the base directory
+        file_path = os.path.join(UPLOAD_DIR, file.name)
+        with open(file_path, "wb") as f:
+            f.write(file.read())
+        
+        # Read the CSV file and return the first few rows as a preview
+        df = pd.read_csv(file_path)
+        return df.head().to_html()
+    except Exception as e:
+        return str(e)
+
+iface = gr.Interface(
+    fn=upload_file,
+    inputs=gr.inputs.File(label="Upload CSV File"),
+    outputs="html",
+    title="CSV File Uploader",
+    description="Upload a CSV file and preview the first few rows."
+)
 
 if __name__ == "__main__":
-    main()
+    iface.launch()