Update src/report_generator.py

src/report_generator.py

@@ -12,12 +12,60 @@ from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, Tabl
 from reportlab.lib import colors
 from reportlab.lib.units import inch, cm
 from reportlab.platypus import PageBreak, KeepTogether
+import requests
 
 from model import orchestrator_chat
 from utils import format_conversation_history
 from session_state import get_full_history
 
 
+# Function to fetch current date and time from an API
+def fetch_current_datetime():
+    """
+    Fetch the current date and time from a public API.
+    Returns a dictionary with date, time, and formatted timestamp.
+    If the API call fails, falls back to system time.
+    """
+    try:
+        # Call the WorldTimeAPI to get current date and time
+        response = requests.get("http://worldtimeapi.org/api/ip")
+        if response.status_code == 200:
+            data = response.json()
+            datetime_str = data.get('datetime')
+            if datetime_str:
+                # Parse the datetime string
+                dt = datetime.fromisoformat(datetime_str.replace('Z', '+00:00'))
+                
+                # Format the date and time
+                formatted_date = dt.strftime("%Y-%m-%d")
+                formatted_time = dt.strftime("%H:%M:%S")
+                full_timestamp = dt.strftime("%Y-%m-%d %H:%M:%S")
+                
+                return {
+                    "date": formatted_date,
+                    "time": formatted_time,
+                    "timestamp": full_timestamp
+                }
+        
+        # Fallback to system time if API fails
+        print("API call failed, using system time instead")
+        now = datetime.now()
+        return {
+            "date": now.strftime("%Y-%m-%d"),
+            "time": now.strftime("%H:%M:%S"),
+            "timestamp": now.strftime("%Y-%m-%d %H:%M:%S")
+        }
+    except Exception as e:
+        # Handle any exceptions and fall back to system time
+        print(f"Error fetching time from API: {str(e)}")
+        now = datetime.now()
+        return {
+            "date": now.strftime("%Y-%m-%d"),
+            "time": now.strftime("%H:%M:%S"),
+            "timestamp": now.strftime("%Y-%m-%d %H:%M:%S")
+        }
+
+
 # Function to encode images to base64 (similar to app.py)
 def get_image_base64(image_path):
     try:
@@ -257,19 +305,23 @@ def build_medical_report(data: dict) -> bytes:
     # --- TABLE 1: Patient Information ---
     patient = data.get('patient', {})
     
+    # Get report generation timestamp
+    generation_time = data.get('generation_timestamp', datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
+    
     # Create a cleaner patient info table with proper Paragraph objects
     patient_data = [
         [Paragraph('<font color="#BA55D3"><b>Patient Information</b></font>', styles['SectionHeading']), ''],
         [Paragraph('Name:', styles['NormalText']), Paragraph(f'{patient.get("name", "–")}', styles['NormalText'])],
         [Paragraph('Age:', styles['NormalText']), Paragraph(f'{patient.get("age", "–")}', styles['NormalText'])],
         [Paragraph('Gender:', styles['NormalText']), Paragraph(f'{patient.get("gender", "–")}', styles['NormalText'])],
-        [Paragraph('Visit Date:', styles['NormalText']), Paragraph(f'{data.get("visit_date", datetime.now().strftime("%Y-%m-%d"))}', styles['NormalText'])]
+        [Paragraph('Visit Date:', styles['NormalText']), Paragraph(f'{data.get("visit_date", datetime.now().strftime("%Y-%m-%d"))}', styles['NormalText'])],
+        [Paragraph('Report Generated:', styles['NormalText']), Paragraph(f'{generation_time}', styles['NormalText'])]
     ]
     
     patient_table = Table(
         patient_data, 
         colWidths=[doc.width * 0.3, doc.width * 0.7],
-        rowHeights=[0.4*inch, 0.25*inch, 0.25*inch, 0.25*inch, 0.25*inch]
+        rowHeights=[0.4*inch, 0.25*inch, 0.25*inch, 0.25*inch, 0.25*inch, 0.25*inch]
     )
     
     patient_table.setStyle(TableStyle([
@@ -478,9 +530,14 @@ def generate_and_download_report():
                 # Update with patient info from the form
                 medical_data["patient"].update(st.session_state.patient_info)
                 
-                # Set the visit date to today if not present
-                if not medical_data.get("visit_date"):
-                    medical_data["visit_date"] = datetime.now().strftime("%Y-%m-%d")
+                # Fetch current date and time from the API
+                datetime_data = fetch_current_datetime()
+                
+                # Set the visit date to today's date from API
+                medical_data["visit_date"] = datetime_data["date"]
+                
+                # Set the report generation timestamp
+                medical_data["generation_timestamp"] = datetime_data["timestamp"]
                 
                 # Generate PDF
                 pdf_data = build_medical_report(medical_data)