Update app.py

app.py

@@ -1,71 +1,139 @@
 import gradio as gr
-import threading
-import time
-import os
-from twilio.rest import Client
+from datetime import datetime, timedelta
 import google.generativeai as genai
+from github import Github, GithubException
+import gitlab
+import docx
+import tempfile
+import requests
 
-# Constants
-TWILIO_SID = os.getenv('TWILIO_SID')
-TWILIO_TOKEN = os.getenv('TWILIO_TOKEN')
-GEMINI_KEY = os.getenv('GEMINI_KEY')
+def generate_release_notes(git_provider, repo_url, personal_access_token, gemini_api_key, start_date, end_date):
+    try:
+        start_date = datetime.strptime(start_date, "%Y-%m-%d")
+        end_date = datetime.strptime(end_date, "%Y-%m-%d")
 
-# Initialize Twilio client
-twilio_client = Client(TWILIO_SID, TWILIO_TOKEN)
+        if git_provider == "GitHub":
+            g = Github(personal_access_token)
+            repo = g.get_repo(repo_url)
+            commits = list(repo.get_commits(since=start_date, until=end_date))
+            commit_messages = [commit.commit.message for commit in commits]
+        elif git_provider == "GitLab":
+            gl = gitlab.Gitlab(url='https://gitlab.com', private_token=personal_access_token)
+            project = gl.projects.get(repo_url)
+            commits = project.commits.list(since=start_date.isoformat(), until=end_date.isoformat())
+            commit_messages = [commit.message for commit in commits]
+        elif git_provider == "Gitea":
+            base_url = "https://gitea.com/api/v1"
+            headers = {"Authorization": f"token {personal_access_token}"}
+            response = requests.get(f"{base_url}/repos/{repo_url}/commits", headers=headers, params={
+                "since": start_date.isoformat(),
+                "until": end_date.isoformat()
+            })
+            response.raise_for_status()
+            commits = response.json()
+            commit_messages = [commit['commit']['message'] for commit in commits]
+        else:
+            return "Unsupported Git provider", None, None
 
-# Initialize Gemini client
-gemini_client = genai.GenerativeModel('gemini-2.0-flash-lite')
+        commit_text = "\n".join(commit_messages)
+        
+        if not commit_text:
+            return "No commits found in the specified date range.", None, None
+        
+        genai.configure(api_key=gemini_api_key)
+        model = genai.GenerativeModel('gemini-2.0-flash-lite')
+        
+        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:
+        - The output should be plain text without any markdown or "-" for post processing
+        - Use section titles followed by a colon (e.g., "New Features:")
+        - Start each item on a new line
+        - Be sure to briefly explain the why and benefits of the change for average users that are non-technical
+        """
 
-# Function to set Dial-in number and Passcode
-def set_call_info(number, passcode):
-    global DIAL_IN_NUMBER, PASSCODE
-    DIAL_IN_NUMBER = f"1{number}"  # Add '1' for US-based numbers
-    PASSCODE = passcode
-    return "Call information set successfully."
+        response = model.generate_content(prompt)
+        release_notes = response.text
+        
+        # Create DOCX file
+        doc = docx.Document()
+        doc.add_heading('Release Notes', 0)
+        
+        current_section = None
+        for line in release_notes.split('\n'):
+            line = line.strip()
+            if line.endswith(':'):
+                doc.add_heading(line, level=1)
+                current_section = None
+            elif line:
+                if current_section is None:
+                    current_section = doc.add_paragraph().style
+                    current_section.name = 'List Bullet'
+                doc.add_paragraph(line, style=current_section)
+        
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.docx') as temp_docx:
+            doc.save(temp_docx.name)
+            docx_path = temp_docx.name
 
-# Function for live transcription using Gemini
-def live_transcription():
-    # Simulate a long call with live transcription using Gemini
-    transcription_chunks = []
-    for i in range(10):  # Simulate 10 chunks
-        chunk = gemini_client.generate_text(f"Transcription chunk {i}")
-        transcription_chunks.append(chunk)
-        time.sleep(1)  # Simulate processing time
-    return transcription_chunks
+        # Create Markdown file
+        markdown_content = "# Release Notes\n\n"
+        for line in release_notes.split('\n'):
+            line = line.strip()
+            if line.endswith(':'):
+                markdown_content += f"\n## {line}\n\n"
+            elif line:
+                markdown_content += f"- {line}\n"
 
-# Function to stitch transcription chunks
-def stitch_transcription_chunks(chunks):
-    return ' '.join(chunks)
-
-# Function to generate meeting minutes using Gemini
-def generate_meeting_minutes(transcript):
-    return gemini_client.generate_text(f"Meeting minutes based on: {transcript}")
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.md', mode='w', encoding='utf-8') as temp_md:
+            temp_md.write(markdown_content)
+            md_path = temp_md.name
+        
+        return release_notes, docx_path, md_path
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}", None, None
 
-# Function to download transcript or meeting minutes as Word document
-def download_as_word(text, filename):
-    doc = Document()
-    doc.add_paragraph(text)
-    doc.save(filename)
-    return filename
+default_end_date = datetime.now()
+default_start_date = default_end_date - timedelta(days=30)
 
-# Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("## Call Transcription and Meeting Minutes Generator")
-    
-    with gr.Row():
-        dial_in_number = gr.Textbox(label="Dial-in Number (without '1')")
-        passcode = gr.Textbox(label="Passcode")
-        set_call_button = gr.Button("Set Call Information")
-        set_call_button.click(set_call_info, inputs=[dial_in_number, passcode], outputs=gr.Textbox(label="Status"))
-    
-    with gr.Row():
-        hang_up_button = gr.Button("Hang Up")
-        hang_up_button.click(lambda: "Call ended", outputs=gr.Textbox(label="Status"))
-    
-    with gr.Row():
-        transcription_output = gr.Textbox(label="Live Transcription Output")
-        minutes_output = gr.Textbox(label="Meeting Minutes Output")
-        download_button = gr.Button("Download Transcript and Minutes")
-        download_button.click(lambda: download_as_word(f"{transcription_output.value}\n{minutes_output.value}", "output.docx"), outputs=gr.File(label="Download"))
+iface = gr.Interface(
+    fn=generate_release_notes,
+    inputs=[
+        gr.Dropdown(
+            choices=["GitHub", "GitLab", "Gitea"],
+            label="Git Provider"
+        ),
+        gr.Textbox(label="Repository URL", placeholder="owner/repo"),
+        gr.Textbox(label="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 (DOCX)"),
+        gr.File(label="Download Release Notes (Markdown)")
+    ],
+    title="Automated Release Notes Generator",
+    description="Generate release notes based on Git commits using Gemini AI. Select a Git provider, enter repository details, and specify the date range for commits.",
+    allow_flagging="never",
+    theme="default",
+    analytics_enabled=False,
+)
 
-demo.launch(share=True)
+iface.launch()