import osimport jsonimport timeimport requestsfrom dotenv import load_dotenvfrom google import genaifrom datetime import datetime # ANSI color codesclass Colors:    CYAN = '\033[96m'    YELLOW = '\033[93m'    GREEN = '\033[92m'    RED = '\033[91m'    MAGENTA = '\033[95m'    BLUE = '\033[94m'    BOLD = '\033[1m'    UNDERLINE = '\033[4m'    RESET = '\033[0m' # Emojis for different sectionsclass Emojis:    GITHUB = "🐙"    STATS = "📊"    CALENDAR = "📅"    SKILLS = "💻"    STAR = "⭐"    ROCKET = "🚀"    CHART = "📈"    BULB = "💡"    WARNING = "⚠️"    CHECK = "✅"    FIRE = "🔥"    BOOK = "📚"    TOOLS = "🛠️"    GRAPH = "📊"    TARGET = "🎯" # Load environment variablesload_dotenv() # Initialize clientsclient = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))firecrawl_api_key = os.getenv("FIRECRAWL_API_KEY") if not firecrawl_api_key:    print(f"{Colors.RED}{Emojis.WARNING} Warning: FIRECRAWL_API_KEY not found in environment variables{Colors.RESET}") def print_header(text, emoji, color=Colors.BLUE):    """Print a formatted section header with emoji."""    width = 70    print("\n" + "═" * width)    print(f"{color}{Colors.BOLD}{emoji}  {text.center(width-4)}  {emoji}{Colors.RESET}")    print("═" * width + "\n") def print_section(title, content, emoji):    """Print a formatted section with title, content, and emoji."""    print(f"\n{Colors.CYAN}{Colors.BOLD}{emoji} {title}{Colors.RESET}")    print(f"{content}") def poll_extraction_result(extraction_id, api_key, interval=2, max_attempts=15):    """Poll Firecrawl API for extraction results with shorter intervals."""    url = f"https://api.firecrawl.dev/v1/extract/{extraction_id}"    headers = {'Authorization': f'Bearer {api_key}'}     print(f"{Colors.YELLOW}Processing profile data...{Colors.RESET}")     for attempt in range(max_attempts):        try:            response = requests.get(url, headers=headers, timeout=10)            data = response.json()             if data.get('success') and data.get('data'):                print(f"{Colors.GREEN}Data extracted successfully!{Colors.RESET}")                return data['data']            elif data.get('success'):                if attempt % 3 == 0:  # Print progress less frequently                    print(".", end="", flush=True)                time.sleep(interval)            else:                print(f"\n{Colors.RED}API Error: {data.get('error', 'Unknown error')}{Colors.RESET}")                return None         except requests.exceptions.Timeout:            print(f"\n{Colors.RED}Request timed out. Retrying...{Colors.RESET}")            continue        except Exception as e:            print(f"\n{Colors.RED}Error polling results: {e}{Colors.RESET}")            return None     print(f"\n{Colors.RED}Extraction timed out after {max_attempts} attempts.{Colors.RESET}")    return None def extract_github_profile(username, api_key):    """Extract GitHub profile data using Firecrawl with optimized settings."""    if not api_key:        print(f"{Colors.RED}Error: Firecrawl API key is missing{Colors.RESET}")        return None     headers = {        'Content-Type': 'application/json',        'Authorization': f'Bearer {api_key}'    }     github_url = f"https://github.com/{username}"        # Simplified prompt for faster extraction    payload = {        "urls": [github_url],        "prompt": """Extract key GitHub profile data:        - Basic profile information (company, location, bio)        - Repository list and details        - Contribution statistics        - Recent activity        - Social stats""",        "enableWebSearch": False    }     try:        print(f"{Colors.YELLOW}Starting extraction for: {username}{Colors.RESET}")        response = requests.post(            "https://api.firecrawl.dev/v1/extract",            headers=headers,            json=payload,            timeout=15        )         if response.status_code != 200:            print(f"{Colors.RED}API Error ({response.status_code}): {response.text}{Colors.RESET}")            return None         data = response.json()        if not data.get('success'):            print(f"{Colors.RED}API Error: {data.get('error', 'Unknown error')}{Colors.RESET}")            return None         extraction_id = data.get('id')        if not extraction_id:            print(f"{Colors.RED}No extraction ID received{Colors.RESET}")            return None         return poll_extraction_result(extraction_id, api_key)     except requests.exceptions.Timeout:        print(f"{Colors.RED}Initial request timed out{Colors.RESET}")        return None    except Exception as e:        print(f"{Colors.RED}Extraction failed: {e}{Colors.RESET}")        return None def analyze_with_gemini(profile_data, username):    """Use Gemini to analyze GitHub profile data with focus on comprehensive insights."""    prompt = f"""    Analyze this GitHub profile and provide detailed insights from the available data.    Focus on concrete information and metrics.        Structure your response in these sections:    1. Professional Background    - Current company/organization (if available)    - Role/position (if available)    - Professional website or blog links    - Location (if available)     2. Activity Analysis    - Total repositories and forks    - Most active repositories (top 3)    - Contribution frequency    - Recent activity trends    - Streak information     3. Technical Portfolio    - Primary programming languages    - Most used technologies/frameworks    - Top contributed repositories    - Notable project themes     4. Community Engagement    - Followers and following count    - Public contributions    - Pull requests and issues    - Project collaborations     Rules:    - Include only verifiable information from the profile    - List specific repository names and their purposes    - Include contribution statistics where available    - Focus on recent activity (last 6 months)    - Skip sections only if completely unavailable        Profile Data: {json.dumps(profile_data, indent=2)}    """     try:        response = client.models.generate_content(            model="gemini-2.0-flash",            contents=prompt        )                # Clean up response        analysis = response.text.strip()        analysis_lines = [line for line in analysis.split('\n')                          if not any(word in line.lower()                                   for word in ['undetermined', 'unknown', 'limited',                                              'not available', 'needs', 'requires', 'unclear'])]        cleaned_analysis = '\n'.join(line for line in analysis_lines if line.strip())                return format_report(cleaned_analysis, username)     except Exception as e:        print(f"{Colors.RED}Analysis failed: {e}{Colors.RESET}")        return None def format_report(raw_analysis, username):    """Format the analysis into a clean, professional report."""    report = f"""{Colors.BOLD}GitHub Profile Analysis: {username}{Colors.RESET}{Colors.CYAN}{'─' * 40}{Colors.RESET}\n"""     sections = raw_analysis.split('\n')    current_section = None     for line in sections:        line = line.strip()        if not line:            continue         if any(section in line for section in ["Professional Background", "Activity Analysis",                                              "Technical Portfolio", "Community Engagement"]):            report += f"\n{Colors.BOLD}{Colors.BLUE}{line}{Colors.RESET}\n"        elif line.startswith('-'):            report += f"• {line[1:].strip()}\n"        elif line and not line.startswith(('#', '•')):            report += f"  {line}\n"     return report def save_report(report, username):    """Save the report to a file."""    filename = f"github_analysis_{username}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt"    try:        with open(filename, 'w', encoding='utf-8') as f:            # Strip ANSI color codes when saving to file            clean_report = report            for color in vars(Colors).values():                if isinstance(color, str) and color.startswith('\033'):                    clean_report = clean_report.replace(color, '')            f.write(clean_report)        return filename    except Exception as e:        print(f"{Colors.RED}{Emojis.WARNING} Error saving report: {e}{Colors.RESET}")        return None def main():    username = input(f"{Colors.GREEN}GitHub username: {Colors.RESET}").strip()     if not username:        print(f"{Colors.RED}Please provide a valid username.{Colors.RESET}")        return     print("Analyzing profile...")    profile_data = extract_github_profile(username, firecrawl_api_key)        if not profile_data:        print(f"{Colors.RED}Profile analysis failed.{Colors.RESET}")        return     report = analyze_with_gemini(profile_data, username)     if report:        print(report)    else:        print(f"{Colors.RED}Could not generate insights.{Colors.RESET}") if __name__ == "__main__":    main()