Update sheets_integration.py

sheets_integration.py

@@ -41,33 +41,57 @@ class SheetsLogger:
             self.gc = None
             self.sheet = None
 
-    def log_request(self, request_data):
+    def initialize_sheet(self):
+        """Initialize sheet with headers if needed"""
+        headers = [
+            "Timestamp",
+            "Name",
+            "Employee ID",
+            "Email",
+            "Department",
+            "Request Details",
+            "Frequency",
+            "Urgency",
+            "User Summary",
+            "Technical Analysis",
+            "Status",  # For tracking request status
+            "Notes"    # For additional comments
+        ]
+        
+        try:
+            # Check if headers exist
+            existing_headers = self.sheet.row_values(1)
+            if not existing_headers:
+                self.sheet.insert_row(headers, 1)
+        except Exception as e:
+            print(f"Error initializing headers: {str(e)}")
+    
+    def log_request(self, data):
         """Log a request to Google Sheets"""
         if not self.sheet:
-            print("Sheet not initialized")
             return False
             
         try:
-            # Prepare row data
             row = [
-                datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  # Timestamp
-                request_data.get('name', ''),
-                request_data.get('employee_id', ''),
-                request_data.get('email', ''),
-                request_data.get('department', ''),
-                request_data.get('request_details', ''),
-                request_data.get('frequency', ''),
-                request_data.get('urgency', ''),
-                request_data.get('system_response', '')
+                data.get('timestamp', ''),
+                data.get('name', ''),
+                data.get('employee_id', ''),
+                data.get('email', ''),
+                data.get('department', ''),
+                data.get('request_details', ''),
+                data.get('frequency', ''),
+                data.get('urgency', ''),
+                data.get('user_summary', ''),
+                data.get('system_analysis', ''),
+                'New',  # Initial status
+                ''     # Empty notes
             ]
             
-            # Append row to sheet
             self.sheet.append_row(row)
             return True
             
         except Exception as e:
             print(f"Error logging to sheet: {str(e)}")
-            return False
 
     def get_all_requests(self):
         """Get all requests from the sheet"""