|
|
import subprocess |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
BINARY_PATH = "./bin/llama-cli" |
|
|
MODEL_PATH = "./qwen0.5-finetuned.gguf" |
|
|
|
|
|
|
|
|
commit_prompt = """Generate a meaningful commit message explaining all the changes in the provided Git diff. |
|
|
|
|
|
### Git Diff: |
|
|
{} |
|
|
|
|
|
### Commit Message: |
|
|
""" |
|
|
|
|
|
|
|
|
git_diff_example = """ |
|
|
diff --git a/index.html b/index.html |
|
|
index 89abcde..f123456 100644 |
|
|
--- a/index.html |
|
|
+++ b/index.html |
|
|
@@ -5,16 +5,6 @@ <body> |
|
|
<h1>Welcome to My Page</h1> |
|
|
|
|
|
- <table border="1"> |
|
|
- <tr> |
|
|
- <th>Name</th> |
|
|
- <th>Age</th> |
|
|
- </tr> |
|
|
- <tr> |
|
|
- <td>John Doe</td> |
|
|
- <td>30</td> |
|
|
- </tr> |
|
|
- </table> |
|
|
|
|
|
+ <p>This is a newly added paragraph replacing the table.</p> |
|
|
</body> |
|
|
</html> |
|
|
""" |
|
|
|
|
|
def generate_commit(git_diff: str, max_tokens: int = 64) -> str: |
|
|
"""Generate a commit message using the llama-cli binary.""" |
|
|
|
|
|
if not git_diff.strip(): |
|
|
return "Please provide a git diff to summarize." |
|
|
|
|
|
prompt_text = commit_prompt.format(git_diff) |
|
|
|
|
|
cmd = [ |
|
|
BINARY_PATH, |
|
|
"-m", MODEL_PATH, |
|
|
"-p", prompt_text, |
|
|
"-n", str(max_tokens) |
|
|
] |
|
|
|
|
|
try: |
|
|
|
|
|
result = subprocess.run(cmd, capture_output=True, text=True, check=True) |
|
|
return result.stdout.strip() |
|
|
except subprocess.CalledProcessError as e: |
|
|
return f"Error running binary: {e}\n{e.stderr}" |
|
|
|
|
|
if __name__ == "__main__": |
|
|
demo = gr.Interface( |
|
|
fn=generate_commit, |
|
|
inputs=[ |
|
|
gr.Textbox(lines=30, label="Git Diff", value=git_diff_example), |
|
|
gr.Slider(1, 2048, value=64, step=1, label="max_tokens") |
|
|
], |
|
|
outputs=gr.Textbox(label="Commit Message", lines=8), |
|
|
title="Commit Message Generator", |
|
|
description="Paste a git diff and generate a concise commit message using the GGUF model via llama-cli binary.", |
|
|
allow_flagging="never" |
|
|
) |
|
|
demo.launch(share=False) |
|
|
|
