Update app.py

app.py

@@ -19,7 +19,7 @@ def generate_release_notes(github_repo, 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.", None
+            return "No commits found in the specified date range.", None, None
         
         genai.configure(api_key=gemini_api_key)
         model = genai.GenerativeModel('gemini-2.5-pro-preview-03-25')
@@ -37,12 +37,11 @@ def generate_release_notes(github_repo, github_token, gemini_api_key, start_date
         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, do not use any bullet points or numbers. Start each item directly with the text.
-        - Keep the text structure simple and easy to convert to a Word document format
-        - be sure to briefly explain the why and benefits of the change for average user that are non technical
+        - Use proper Markdown syntax
+        - Use # for main headers and ## for subheaders
+        - Use - for list items
+        - Keep the text structure simple and easy to read
+        - Be sure to briefly explain the why and benefits of the change for average users that are non-technical
         """
 
         response = model.generate_content(prompt)
@@ -54,30 +53,35 @@ def generate_release_notes(github_repo, github_token, gemini_api_key, start_date
         current_list_style = None
         for line in release_notes.split('\n'):
             line = line.strip()
-            if line.endswith(':'):
-                doc.add_heading(line, level=1)
+            if line.startswith('# '):
+                doc.add_heading(line[2:], level=1)
                 current_list_style = None
-            elif line:
+            elif line.startswith('## '):
+                doc.add_heading(line[3:], level=2)
+                current_list_style = None
+            elif line.startswith('- '):
                 if current_list_style is None:
                     current_list_style = doc.add_paragraph().style
                     current_list_style.name = 'List Bullet'
                     current_list_style.font.size = docx.shared.Pt(11)
-                doc.add_paragraph(line, style=current_list_style)
+                doc.add_paragraph(line[2:], style=current_list_style)
+            elif line:
+                doc.add_paragraph(line)
         
         temp_file = "release_notes.docx"
         doc.save(temp_file)
         
-        return release_notes, temp_file
+        return release_notes, temp_file, release_notes
     
     except GithubException as e:
         if e.status == 401:
-            return "Error: Invalid GitHub token or insufficient permissions.", None
+            return "Error: Invalid GitHub token or insufficient permissions.", None, None
         elif e.status == 404:
-            return f"Error: Repository not found. Please check the GitHub repository name. Attempted to access: {github_repo}", None
+            return f"Error: Repository not found. Please check the GitHub repository name. Attempted to access: {github_repo}", None, None
         else:
-            return f"GitHub API error: {str(e)}", None
+            return f"GitHub API error: {str(e)}", None, None
     except Exception as e:
-        return f"An error occurred: {str(e)}", None
+        return f"An error occurred: {str(e)}", None, None
 
 default_end_date = datetime.now()
 default_start_date = default_end_date - timedelta(days=30)
@@ -101,7 +105,8 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Generated Release Notes"),
-        gr.File(label="Download Release Notes")
+        gr.File(label="Download Release Notes (DOCX)"),
+        gr.File(label="Download Release Notes (Markdown)", file_types=[".md"])
     ],
     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.",