import json
import os
from datetime import datetime
import glob
from app import SubmissionForum
from utils.hub_storage import HubStorage



def read_json_file(file_path: str) -> dict:
    """Read and parse local JSON file."""
    try:
        with open(file_path, 'r') as f:
            return json.load(f)
    except Exception as e:
        print(f"Error reading {file_path}: {str(e)}")
        return None

def update_json_file(file_path: str, content: dict, method_name: str = None, new_status: str = None) -> bool:
    """Update local JSON file and add forum post if status changed"""
    try:
        with open(file_path, 'w') as f:
            json.dump(content, f, indent=4)
            
        # Add forum post if this is a status update
        if method_name and new_status:
            forum.add_status_update(method_name, new_status)
            
        return True
    except Exception as e:
        print(f"Error updating {file_path}: {str(e)}")
        return False

def review_pending_submissions(submissions_dir: str = "submissions"):
    """Review all pending submissions and allow admin to approve/reject them."""
    try:
        REPO_ID = "snap-stanford/stark-leaderboard" 
        hub_storage = HubStorage(REPO_ID)
        forum = SubmissionForum(hub_storage=hub_storage)
        # Find all latest.json files
        latest_files = glob.glob(os.path.join(submissions_dir, "**", "latest.json"), recursive=True)
        
        if not latest_files:
            print("No submissions found.")
            return
        
        changes_made = False
        
        # Process each submission
        for latest_file in latest_files:
            folder_path = os.path.dirname(latest_file)
            latest_info = read_json_file(latest_file)
            
            if not latest_info:
                print(f"Could not read {latest_file}, skipping...")
                continue
                
            if latest_info.get('status') != 'pending_review':
                continue
                
            # Find corresponding metadata file
            timestamp = latest_info.get('latest_submission')
            if not timestamp:
                print(f"No timestamp found in {latest_file}, skipping...")
                continue
                
            metadata_file = os.path.join(folder_path, f"metadata_{timestamp}.json")
            metadata = read_json_file(metadata_file)
            
            if not metadata:
                print(f"Could not read metadata for {latest_file}, skipping...")
                continue
            
            # Display submission details
            print("\n" + "="*80)
            print(f"Submission Details:")
            print(f"Method Name: {metadata.get('Method Name')}")
            print(f"Team Name: {metadata.get('Team Name')}")
            print(f"Dataset: {metadata.get('Dataset')}")
            print(f"Split: {metadata.get('Split')}")
            print(f"Contact Email(s): {metadata.get('Contact Email(s)')}")
            print(f"Code Repository: {metadata.get('Code Repository')}")
            print("\nModel Description:")
            print(metadata.get('Model Description', 'No description provided'))
            print("\nResults:")
            if 'results' in metadata:
                for metric, value in metadata['results'].items():
                    print(f"{metric}: {value}")
            print("\nSubmission Date:", metadata.get('submission_date', 'Unknown'))
            print("="*80)
            
            # Get admin decision
            while True:
                decision = input("\nApprove this submission? (y/n/skip/quit): ").lower()
                if decision in ['y', 'n', 'skip', 'quit']:
                    break
                print("Invalid input. Please enter 'y', 'n', 'skip', or 'quit'")
            
            if decision == 'quit':
                print("Exiting review process...")
                break
                
            if decision == 'skip':
                continue
                
            # Update status
            new_status = 'approved' if decision == 'y' else 'rejected'
            review_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
            
            # Update latest.json
            latest_info['status'] = new_status
            latest_info['reviewed_at'] = review_timestamp
            if not update_json_file(
                latest_file, 
                latest_info, 
                method_name=metadata.get('Method Name'),
                new_status=new_status
            ):
                print("Failed to update latest.json")
                continue
            
            # Update metadata
            metadata['status'] = new_status
            metadata['reviewed_at'] = review_timestamp
            if not update_json_file(metadata_file, metadata):
                print("Failed to update metadata file")
                continue
            
            forum.add_status_update(metadata.get('Method Name'), new_status)
            print(f"Submission {new_status}")
            changes_made = True
            
        if changes_made:
            print("\nChanges have been made to the following files:")
            print("Please push these changes to HuggingFace manually.")
            
    except Exception as e:
        print(f"Error during review process: {str(e)}")

if __name__ == "__main__":
    print("Starting submission review process...")
    # You can specify a different submissions directory if needed
    review_pending_submissions("submissions")
    print("\nReview process completed.")