Update main.py

main.py

@@ -1062,8 +1062,8 @@ def test_agent():
 @app.route('/api/projects/<project_id>/log-call-usage', methods=['POST'])
 def log_call_usage(project_id):
     """
-    Calculates and deducts credits from a user's account in Firebase
-    after a call is completed.
+    ✅ MODIFIED: Now accepts and stores the full conversation transcript
+    in addition to calculating credit cost.
     """
     logger.info(f"[LOGGING] Received usage log for project: {project_id}")
     
@@ -1073,32 +1073,45 @@ def log_call_usage(project_id):
 
     data = request.get_json()
     duration_seconds = data.get("durationSeconds")
+    transcript = data.get("transcript") # Get the new transcript field
 
-    if duration_seconds is None or not isinstance(duration_seconds, (int, float)):
+    if duration_seconds is None:
         return jsonify({'error': 'Invalid duration provided.'}), 400
 
-    # Calculate credit cost (3 credits per minute, always rounded up)
+    # --- Credit Calculation ---
     minutes = math.ceil(duration_seconds / 60)
     cost = minutes * 3
-    
-    logger.info(f"[LOGGING] User '{uid}' call duration: {duration_seconds:.2f}s, rounded to {minutes} minute(s). Cost: {cost} credits.")
+    logger.info(f"[LOGGING] User '{uid}' call duration: {duration_seconds:.2f}s, Cost: {cost} credits.")
 
     try:
+        # --- Transcript Storage ---
+        if transcript and isinstance(transcript, str) and len(transcript) > 10:
+            transcript_id = f"{project_id}_{int(time.time())}"
+            transcript_ref = db_ref.child(f'transcripts/{uid}/{transcript_id}')
+            transcript_data = {
+                "transcript": transcript,
+                "projectId": project_id,
+                "userId": uid,
+                "durationSeconds": duration_seconds,
+                "createdAt": datetime.utcnow().isoformat()
+            }
+            transcript_ref.set(transcript_data)
+            logger.info(f"[LOGGING] Successfully stored transcript {transcript_id} for user '{uid}'.")
+        else:
+            logger.warning(f"[LOGGING] No valid transcript provided for user '{uid}' on project {project_id}.")
+
+        # --- Credit Deduction ---
         user_ref = db_ref.child(f'users/{uid}')
         user_data = user_ref.get()
 
         if user_data is None:
-            logger.error(f"[LOGGING] User with UID '{uid}' not found in the database.")
             return jsonify({'error': 'User not found.'}), 404
 
         current_credits = user_data.get('credits', 0)
-        
         new_credits = max(0, current_credits - cost)
-        
         user_ref.update({'credits': new_credits})
         
-        logger.info(f"[LOGGING] Successfully updated credits for user '{uid}'. Old: {current_credits}, New: {new_credits}")
-
+        logger.info(f"[LOGGING] Updated credits for user '{uid}'. New balance: {new_credits}")
         return jsonify({
             "status": "success",
             "creditsDeducted": cost,
@@ -1110,6 +1123,7 @@ def log_call_usage(project_id):
         return jsonify({'error': 'A server error occurred while updating credits.'}), 500
 
 
+
 # -----------------------------------------------------------------------------
 # 7. MAIN EXECUTION
 # -----------------------------------------------------------------------------