Update interview_history.py

interview_history.py

@@ -1,4 +1,6 @@
 # PrepGenie/interview_history.py
+"""Handles saving and loading interview history to/from Firestore."""
+
 import firebase_admin
 from firebase_admin import firestore
 import datetime
@@ -8,116 +10,113 @@ import json
 # Assumes Firebase Admin is initialized in app.py
 try:
     db = firestore.client()
-    print("Firestore client for history initialized successfully.")
+    print("Firestore client for history initialized successfully in interview_history.")
     FIRESTORE_AVAILABLE = True
 except Exception as e:
-    print(f"Error initializing Firestore client for history: {e}")
+    print(f"Error initializing Firestore client for history in interview_history: {e}")
     db = None
     FIRESTORE_AVAILABLE = False
 
 def save_interview_history(user_id, interview_data):
     """
     Saves the mock interview results to Firestore under the user's document.
-
-    Args:
-        user_id (str): The unique ID of the logged-in user (from Firebase Auth).
-        interview_data (dict): A dictionary containing interview details.
-            Expected structure:
-            {
-                "timestamp": str (ISO format datetime),
-                "resume_overview": str,
-                "selected_roles": list,
-                "questions": list,
-                "answers": list,
-                "feedback": list,
-                "interactions": dict,
-                "metrics_list": list,
-                "final_metrics": dict,
-                "average_rating": float,
-                "evaluation_report": str,
-                # ... other relevant data ...
-            }
-
-    Returns:
-        bool: True if successful, False otherwise.
     """
     if not FIRESTORE_AVAILABLE or not db:
-        print("Firestore client not available. Cannot save history.")
+        print("Firestore client not available in interview_history. Cannot save history.")
         return False
     if not user_id:
-        print("User ID is required to save history.")
+        print("User ID is required in interview_history to save history.")
         return False
 
     try:
-        # Reference to the user's document in the 'users' collection
         user_ref = db.collection('users').document(user_id)
-
-        # Add the interview data to a subcollection 'interview_history'
         history_ref = user_ref.collection('interview_history')
-
-        # Add the data with an auto-generated ID
         history_ref.add(interview_data)
-
-        print(f"Interview history saved for user {user_id}")
+        print(f"Interview history saved for user {user_id} in interview_history")
         return True
     except Exception as e:
-        print(f"Error saving interview history for user {user_id}: {e}")
+        print(f"Error saving interview history for user {user_id} in interview_history: {e}")
         return False
 
 def load_interview_history(user_id, limit=5):
     """
     Loads the mock interview history for a specific user from Firestore.
-
-    Args:
-        user_id (str): The unique ID of the logged-in user.
-        limit (int): The maximum number of recent interviews to retrieve.
-
-    Returns:
-        list: A list of dictionaries, each representing an interview record.
-              Returns an empty list if no history is found or on error.
     """
     if not FIRESTORE_AVAILABLE or not db:
-        print("Firestore client not available. Cannot load history.")
+        print("Firestore client not available in interview_history. Cannot load history.")
         return []
     if not user_id:
-        print("User ID is required to load history.")
+        print("User ID is required in interview_history to load history.")
         return []
 
     try:
-        # Reference to the user's interview history subcollection
         history_ref = db.collection('users').document(user_id).collection('interview_history')
-
-        # Query the history, ordered by timestamp descending (most recent first), limited
         docs = history_ref.order_by('timestamp', direction=firestore.Query.DESCENDING).limit(limit).stream()
 
         history_list = []
         for doc in docs:
-            # Convert Firestore document to dictionary
             interview_record = doc.to_dict()
-            # Add the document ID if needed (optional)
-            # interview_record['doc_id'] = doc.id
             history_list.append(interview_record)
 
-        print(f"Loaded {len(history_list)} interview records for user {user_id}")
+        print(f"Loaded {len(history_list)} interview records for user {user_id} in interview_history")
         return history_list
     except Exception as e:
-        print(f"Error loading interview history for user {user_id}: {e}")
+        print(f"Error loading interview history for user {user_id} in interview_history: {e}")
         return []
 
-# Example usage (for testing or understanding structure):
-# Assuming you have a user ID and interview data after an interview session:
-# user_uid = "some_user_id_from_firebase_auth"
-# interview_summary_data = {
-#     "timestamp": datetime.datetime.now().isoformat(),
-#     "resume_overview": "Overview of the resume...",
-#     "selected_roles": ["Software Engineer"],
-#     "questions": ["Question 1?", "Question 2?"],
-#     "answers": ["Answer 1", "Answer 2"],
-#     "feedback": ["Feedback 1", "Feedback 2"],
-#     # ... other fields ...
-# }
-# save_interview_history(user_uid, interview_summary_data)
+# --- History Loading Function for UI ---
+def load_user_history(user_id_state):
+    """Function to load and format interview history for display in the UI."""
+    if not FIRESTORE_AVAILABLE:
+        return "History feature is not available (Firestore not initialized)."
+    if not user_id_state:
+        return "Please log in to view your interview history."
 
-# Load last 3 interviews
-# past_interviews = load_interview_history(user_uid, limit=3)
-# print(past_interviews)
+    try:
+        history_list = load_interview_history(user_id_state, limit=5)
+
+        if not history_list:
+            return "No interview history found."
+
+        output_text = "**Your Recent Mock Interviews:**\n\n"
+        for idx, record in enumerate(history_list):
+            timestamp = record.get('timestamp', 'Unknown Time')
+            try:
+                if 'Z' in timestamp:
+                    dt = datetime.datetime.fromisoformat(timestamp.replace('Z', '+00:00'))
+                else:
+                    dt = datetime.datetime.fromisoformat(timestamp)
+                formatted_time = dt.strftime("%Y-%m-%d %H:%M:%S")
+            except Exception as e:
+                print(f"Error parsing timestamp {timestamp} in load_user_history: {e}")
+                formatted_time = timestamp
+
+            roles = ", ".join(record.get('selected_roles', ['N/A']))
+            avg_rating = record.get('average_rating', 'N/A')
+            output_text += f"--- **Interview #{len(history_list) - idx} ({formatted_time})** ---\n"
+            output_text += f"**Roles Applied:** {roles}\n"
+            output_text += f"**Average Rating:** {avg_rating:.2f}\n\n"
+
+            interactions = record.get('interactions', {})
+            if interactions:
+                output_text += "**Interview Snippets:**\n"
+                count = 0
+                for q_key, a_val in list(interactions.items())[:3]:
+                    q_display = q_key.split(':', 1)[1].strip() if ':' in q_key else q_key
+                    a_display = a_val.split(':', 1)[1].strip() if ':' in a_val else a_val
+                    output_text += f"- **Q:** {q_display[:100]}{'...' if len(q_display) > 100 else ''}\n"
+                    output_text += f"  **A:** {a_display[:100]}{'...' if len(a_display) > 100 else ''}\n"
+                    count += 1
+                    if count >= 3:
+                        break
+                if len(interactions) > 3:
+                    output_text += f"... (and {len(interactions) - 3} more questions)\n"
+            else:
+                 output_text += "**Details:** Not available.\n"
+            output_text += "\n---\n\n"
+
+        return output_text
+    except Exception as e:
+        error_msg = f"Error loading interview history in load_user_history: {str(e)}"
+        print(error_msg)
+        return error_msg