Update app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
 import requests
+import matplotlib.pyplot as plt
 
-# Function to fetch repository details from GitHub
+# 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:
@@ -36,26 +37,39 @@ def analyze_github_repo(repo_url):
         contributors_data = requests.get(contributors_url).json()
         contributors = len(contributors_data)
 
-    # Return the analysis result
-    result = (
-        f"Repository: {repo_url}\n"
-        f"Stars: {stars}\n"
-        f"Forks: {forks}\n"
-        f"Open Issues: {issues}\n"
-        f"Contributors: {contributors}"
-    )
-    
-    return result
+    # 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.Textbox(label="Repository Analysis", placeholder="Analysis will be displayed here..."),
+    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."
+    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()
 
+