import gradio as gr
import requests
import matplotlib.pyplot as plt

# Function to fetch repository details from GitHub and generate a visualization
def analyze_github_repo(repo_url):
    # Extract the username and repo name from the GitHub URL
    if "github.com" not in repo_url:
        return "Please provide a valid GitHub repository URL."

    parts = repo_url.split('/')
    if len(parts) < 5:
        return "URL should be in the format: https://github.com/username/repository"

    user, repo = parts[3], parts[4]

    # GitHub API endpoint to get repository details
    api_url = f"https://api.github.com/repos/{user}/{repo}"
    response = requests.get(api_url)

    # Check if repository exists
    if response.status_code == 404:
        return "Repository not found."
    
    # Parse the response
    data = response.json()
    
    # Extract relevant information
    stars = data.get('stargazers_count', 'N/A')
    forks = data.get('forks_count', 'N/A')
    issues = data.get('open_issues_count', 'N/A')
    contributors_url = data.get('contributors_url', None)

    # Get number of contributors
    contributors = "N/A"
    if contributors_url:
        contributors_data = requests.get(contributors_url).json()
        contributors = len(contributors_data)

    # Data to plot
    metrics = {
        'Stars': stars,
        'Forks': forks,
        'Open Issues': issues,
        'Contributors': contributors
    }

    # Create the bar plot
    fig, ax = plt.subplots()
    ax.bar(metrics.keys(), metrics.values(), color=['blue', 'green', 'red', 'orange'])

    ax.set_xlabel('Metrics')
    ax.set_ylabel('Count')
    ax.set_title(f"GitHub Repository Analysis: {repo_url}")

    # Save the plot as an image
    plt.tight_layout()
    plt.savefig("/mnt/data/github_analysis.png")
    plt.close()

    return "/mnt/data/github_analysis.png"

# Gradio interface for the tool
iface = gr.Interface(
    fn=analyze_github_repo,
    inputs=gr.Textbox(label="Enter GitHub Repository URL", placeholder="https://github.com/username/repository"),
    outputs=gr.Image(label="Repository Analysis Visualization"),
    title="GitHub Repository Analysis Tool",
    description="Enter a GitHub repository URL to get basic analytics including stars, forks, issues, and contributors, along with a visual chart."
)

# Launch the app
iface.launch()