| | import gradio as gr |
| | from datetime import datetime, timedelta |
| | import google.generativeai as genai |
| | from github import Github, GithubException |
| | import docx |
| | import re |
| |
|
| | def generate_release_notes(github_repo, github_token, gemini_api_key, start_date, end_date): |
| | try: |
| | g = Github(github_token) |
| | |
| | repo = g.get_repo(github_repo) |
| | |
| | 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. |
| | |
| | Important formatting instructions: |
| | - Do not use any Markdown syntax (*, #, `, etc.) |
| | - Use plain text only |
| | - For section headers, simply use the text as is (e.g., "New Features:") |
| | - For list items, use a single bullet point (•) followed by a space |
| | - Do not use multiple bullet points or numbers for a single item |
| | - Keep the text structure simple and easy to convert to a Word document format |
| | """ |
| |
|
| | response = model.generate_content(prompt) |
| | release_notes = response.text |
| | |
| | doc = docx.Document() |
| | doc.add_heading('Release Notes', 0) |
| | |
| | for line in release_notes.split('\n'): |
| | line = line.strip() |
| | if line.endswith(':'): |
| | doc.add_heading(line, level=1) |
| | elif line.startswith('•'): |
| | |
| | cleaned_line = re.sub(r'^•+\s*', '• ', line) |
| | doc.add_paragraph(cleaned_line, style='List Bullet') |
| | elif line: |
| | doc.add_paragraph(line) |
| | |
| | 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 repository name. Attempted to access: {github_repo}", 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() |
