bluenevus's picture
Update app.py
a8576c2 verified
raw
history blame
4.54 kB
import gradio as gr
from datetime import datetime, timedelta
import google.generativeai as genai
from github import Github, GithubException
import docx
import re
def remove_links_keep_issue_numbers(text):
return re.sub(r'\[([^\]]+)\]\([^\)]+\)', r'\1', text)
def strip_markdown(text):
# Remove headers
text = re.sub(r'^#+\s*', '', text, flags=re.MULTILINE)
# Remove bold/italic
text = re.sub(r'\*{1,2}([^\*]+)\*{1,2}', r'\1', text)
# Remove inline code
text = re.sub(r'`([^`]+)`', r'\1', text)
# Remove blockquotes
text = re.sub(r'^\s*>\s*', '', text, flags=re.MULTILINE)
return text.strip()
def markdown_to_docx(doc, markdown_text):
lines = markdown_text.split('\n')
for line in lines:
stripped_line = strip_markdown(line)
if line.startswith('# '):
doc.add_heading(stripped_line, level=1)
elif line.startswith('## '):
doc.add_heading(stripped_line, level=2)
elif line.startswith('### '):
doc.add_heading(stripped_line, level=3)
elif line.strip().startswith('- ') or line.strip().startswith('* '):
doc.add_paragraph(stripped_line.strip()[2:], style='List Bullet')
elif re.match(r'^\d+\.', line.strip()):
doc.add_paragraph(stripped_line.strip(), style='List Number')
else:
doc.add_paragraph(stripped_line)
def generate_release_notes(github_url, github_token, gemini_api_key, start_date, end_date):
try:
g = Github(github_token)
repo = g.get_repo(github_url)
start_date = datetime.strptime(start_date, "%Y-%m-%d")
end_date = datetime.strptime(end_date, "%Y-%m-%d")
commits = list(repo.get_commits(since=start_date, until=end_date))
commit_messages = [commit.commit.message for commit in commits]
commit_text = "\n".join(commit_messages)
if not commit_text:
return "No commits found in the specified date range.", None
genai.configure(api_key=gemini_api_key)
model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
prompt = f"""Based on the following commit messages, generate comprehensive release notes:
{commit_text}
Please organize the release notes into sections such as:
1. New Features
2. Bug Fixes
3. Improvements
4. Breaking Changes (if any)
Provide a concise summary for each item. Do not include any links, but keep issue numbers if present."""
response = model.generate_content(prompt)
release_notes = remove_links_keep_issue_numbers(response.text)
doc = docx.Document()
doc.add_heading('Release Notes', 0)
markdown_to_docx(doc, release_notes)
temp_file = "release_notes.docx"
doc.save(temp_file)
return release_notes, temp_file
except GithubException as e:
if e.status == 401:
return "Error: Invalid GitHub token or insufficient permissions.", None
elif e.status == 404:
return f"Error: Repository not found. Please check the GitHub URL. Attempted to access: {github_url}", None
else:
return f"GitHub API error: {str(e)}", None
except Exception as e:
return f"An error occurred: {str(e)}", None
default_end_date = datetime.now()
default_start_date = default_end_date - timedelta(days=30)
iface = gr.Interface(
fn=generate_release_notes,
inputs=[
gr.Textbox(label="GitHub Repository Name", placeholder="MicroHealthLLC/maiko-assistant"),
gr.Textbox(label="GitHub Personal Access Token", type="password"),
gr.Textbox(label="Gemini API Key", type="password"),
gr.Textbox(
label="Start Date",
placeholder="YYYY-MM-DD",
value=default_start_date.strftime("%Y-%m-%d"),
),
gr.Textbox(
label="End Date",
placeholder="YYYY-MM-DD",
value=default_end_date.strftime("%Y-%m-%d"),
)
],
outputs=[
gr.Textbox(label="Generated Release Notes"),
gr.File(label="Download Release Notes")
],
title="Automated Release Notes Generator",
description="Generate release notes based on GitHub commits using Gemini AI. Enter start and end dates (YYYY-MM-DD) to define the time range for commits.",
allow_flagging="never",
theme="default",
analytics_enabled=False,
)
iface.launch()