Update pages/4_View_Patient_Records.py

pages/4_View_Patient_Records.py

@@ -7,25 +7,40 @@ st.title("📁 Stored Patient Reports")
 conn = sqlite3.connect("patients.db")
 c = conn.cursor()
 
-# ✅ Create table if missing
+# --- Ensure base table exists and required columns exist (simple migration) ---
 c.execute("""
 CREATE TABLE IF NOT EXISTS reports (
     id INTEGER PRIMARY KEY AUTOINCREMENT,
-    patient_name TEXT,
-    uploaded_on TEXT,
-    summary TEXT
+    patient_name TEXT
 )
 """)
 conn.commit()
 
-# Now the table definitely exists
-c.execute("SELECT patient_name, uploaded_on, summary FROM reports")
+for col in ("uploaded_on", "summary", "raw_text"):
+    try:
+        c.execute(f"ALTER TABLE reports ADD COLUMN {col} TEXT")
+    except sqlite3.OperationalError:
+        # Column already exists
+        pass
+conn.commit()
+# --------------------------------------------------------------------------
+
+# Now the table definitely exists and has required columns
+c.execute("SELECT patient_name, COALESCE(uploaded_on, ''), COALESCE(summary, '') FROM reports")
 data = c.fetchall()
 
 if len(data) == 0:
     st.info("No reports found.")
 else:
     for name, date, summary in data:
-        st.write(f"### {name} — {date}")
-        st.write(summary)
+        display_date = date if date else "No date"
+        st.write(f"### {name} — {display_date}")
+        if summary:
+            st.write(summary)
+        else:
+            st.info("No summary available for this report.")
         st.write("---")
+
+# Close DB connection
+conn.close()
+