Update app.py

app.py

@@ -1,34 +1,60 @@
 import gradio as gr
 from datetime import datetime, timedelta
+import requests
+import google.generativeai as genai
+from github import Github
 
 def generate_release_notes(github_url, github_token, gemini_api_key, start_date, end_date):
-    # This is a placeholder implementation. In a real scenario, you would use these
-    # parameters to fetch data from GitHub and generate release notes using Gemini AI.
     try:
-        start = datetime.strptime(start_date, "%Y-%m-%d")
-        end = datetime.strptime(end_date, "%Y-%m-%d")
-        days = (end - start).days
+        # Initialize GitHub client
+        g = Github(github_token)
         
-        notes = f"Release Notes for {github_url}\n"
-        notes += f"Period: From {start_date} to {end_date} ({days} days)\n\n"
-        notes += "1. New Features:\n   - Feature A added\n   - Feature B implemented\n\n"
-        notes += "2. Bug Fixes:\n   - Fixed issue with login\n   - Resolved performance problem in module X\n\n"
-        notes += "3. Improvements:\n   - Enhanced user interface\n   - Optimized database queries\n\n"
-        notes += "Note: This is a simulated output. Actual implementation would use GitHub API and Gemini AI."
+        # Extract owner and repo from the URL
+        _, _, _, owner, repo = github_url.rstrip('/').split('/')
         
-        return notes
-    except ValueError:
-        return "Error: Invalid date format. Please use YYYY-MM-DD."
+        # Get the repository
+        repo = g.get_repo(f"{owner}/{repo}")
+        
+        # Get commits between start_date and end_date
+        commits = repo.get_commits(since=datetime.strptime(start_date, "%Y-%m-%d"),
+                                   until=datetime.strptime(end_date, "%Y-%m-%d"))
+        
+        # Prepare commit messages
+        commit_messages = [commit.commit.message for commit in commits]
+        commit_text = "\n".join(commit_messages)
+        
+        # Use Gemini AI to generate release notes
+        genai.configure(api_key=gemini_api_key)
+        model = genai.GenerativeModel('gemini-pro')
+        
+        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."""
+
+        response = model.generate_content(prompt)
+        
+        return response.text
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 # Set default dates
 default_end_date = datetime.now()
-default_start_date = default_end_date - timedelta(days=7)  # One week ago
+default_start_date = default_end_date - timedelta(days=30)  # One week ago
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=generate_release_notes,
     inputs=[
-        gr.Textbox(label="GitHub Repository URL", placeholder="https://github.com/username/repo.git"),
+        gr.Textbox(label="GitHub Repository URL", placeholder="https://github.com/username/repo"),
         gr.Textbox(label="GitHub Personal Access Token", type="password"),
         gr.Textbox(label="Gemini API Key", type="password"),
         gr.Textbox(