Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from datetime import datetime, timedelta
 import google.generativeai as genai
 from github import Github, GithubException
+import docx
 
 def generate_release_notes(github_url, github_token, gemini_api_key, start_date, end_date):
     try:
@@ -21,7 +22,7 @@ def generate_release_notes(github_url, github_token, gemini_api_key, start_date,
         commit_text = "\n".join(commit_messages)
         
         if not commit_text:
-            return "No commits found in the specified date range."
+            return "No commits found in the specified date range.", None
         
         # Use Gemini AI to generate release notes
         genai.configure(api_key=gemini_api_key)
@@ -41,21 +42,30 @@ def generate_release_notes(github_url, github_token, gemini_api_key, start_date,
 
         response = model.generate_content(prompt)
         
-        return response.text
+        # Create a Word document
+        doc = docx.Document()
+        doc.add_heading('Release Notes', 0)
+        doc.add_paragraph(response.text)
+        
+        # Save the document to a temporary file
+        temp_file = "release_notes.docx"
+        doc.save(temp_file)
+        
+        return response.text, temp_file
     
     except GithubException as e:
         if e.status == 401:
-            return "Error: Invalid GitHub token or insufficient permissions."
+            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}"
+            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)}"
+            return f"GitHub API error: {str(e)}", None
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return f"An error occurred: {str(e)}", None
 
 # 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 month ago
 
 # Create Gradio interface
 iface = gr.Interface(
@@ -75,7 +85,10 @@ iface = gr.Interface(
             value=default_end_date.strftime("%Y-%m-%d"),
         )
     ],
-    outputs=gr.Textbox(label="Generated Release Notes"),
+    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",