Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
import os
|
| 3 |
+
from github import Github
|
| 4 |
+
from datetime import datetime
|
| 5 |
+
import google.generativeai as genai
|
| 6 |
+
|
| 7 |
+
def generate_release_notes(github_url, github_token, gemini_api_key, start_date, end_date):
|
| 8 |
+
# Configure Gemini API
|
| 9 |
+
genai.configure(api_key=gemini_api_key)
|
| 10 |
+
model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
|
| 11 |
+
|
| 12 |
+
# Initialize GitHub client
|
| 13 |
+
g = Github(github_token)
|
| 14 |
+
|
| 15 |
+
# Extract repository name from URL
|
| 16 |
+
repo_name = github_url.split('/')[-1].replace('.git', '')
|
| 17 |
+
repo = g.get_repo(f"MicroHealthLLC/{repo_name}")
|
| 18 |
+
|
| 19 |
+
# Convert dates to datetime objects
|
| 20 |
+
start_date = datetime.strptime(start_date, "%Y-%m-%d")
|
| 21 |
+
end_date = datetime.strptime(end_date, "%Y-%m-%d")
|
| 22 |
+
|
| 23 |
+
# Fetch commits
|
| 24 |
+
commits = repo.get_commits(since=start_date, until=end_date)
|
| 25 |
+
|
| 26 |
+
# Prepare commit messages for AI analysis
|
| 27 |
+
commit_messages = [commit.commit.message for commit in commits]
|
| 28 |
+
commit_text = "\n".join(commit_messages)
|
| 29 |
+
|
| 30 |
+
# Generate release notes using Gemini
|
| 31 |
+
prompt = f"Based on the following commit messages, generate comprehensive release notes:\n\n{commit_text}\n\nPlease categorize changes, summarize features, and identify key updates."
|
| 32 |
+
response = model.generate_content(prompt)
|
| 33 |
+
|
| 34 |
+
return response.text
|
| 35 |
+
|
| 36 |
+
# Create Gradio interface
|
| 37 |
+
iface = gr.Interface(
|
| 38 |
+
fn=generate_release_notes,
|
| 39 |
+
inputs=[
|
| 40 |
+
gr.Textbox(label="GitHub Repository URL (e.g., https://github.com/MicroHealthLLC/maiko-assistant.git)"),
|
| 41 |
+
gr.Textbox(label="GitHub Personal Access Token", type="password"),
|
| 42 |
+
gr.Textbox(label="Gemini API Key", type="password"),
|
| 43 |
+
gr.Textbox(label="Start Date (YYYY-MM-DD)"),
|
| 44 |
+
gr.Textbox(label="End Date (YYYY-MM-DD)")
|
| 45 |
+
],
|
| 46 |
+
outputs=gr.Textbox(label="Generated Release Notes"),
|
| 47 |
+
title="Automated Release Notes Generator",
|
| 48 |
+
description="Generate release notes based on GitHub commits using Gemini AI."
|
| 49 |
+
)
|
| 50 |
+
|
| 51 |
+
# Launch the app
|
| 52 |
+
iface.launch()
|