| import gradio as gr |
| from models.codet5 import CodeT5 |
| from models.other_models import OtherModels |
| from repositories.github_api import GitHubAPI |
|
|
| codet5_model = CodeT5() |
| other_models = OtherModels() |
| github_api = GitHubAPI() |
|
|
| def analyze_repository(repo_url): |
| repo_data = github_api.get_repository(repo_url) |
| optimization_results = codet5_model.analyze(repo_data, github_api) |
| bug_hunting_results = other_models.analyze(repo_data, github_api) |
| return optimization_results, bug_hunting_results |
|
|
| iface = gr.Interface( |
| fn=analyze_repository, |
| inputs=gr.Textbox(lines=1, placeholder="Enter GitHub Repository URL (e.g., https://github.com/owner/repo)"), |
| outputs=[ |
| gr.Textbox(lines=10, label="Optimization Results"), |
| gr.Textbox(lines=10, label="Bug Hunting Results"), |
| ], |
| title="GitHub Repository Analyzer", |
| description="Analyze GitHub repositories for optimization suggestions and potential bugs using CodeT5 and other models.", |
| ) |
|
|
| iface.launch() |
