gmsg 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

gmsg/gmsg.py

@@ -4,6 +4,10 @@ from google import genai
 from pathlib import Path
 from .api_key import get_or_set_api_key
 
+GREEN = "\033[92m"
+RED = "\033[91m"
+RESET = "\033[0m"
+
 
 def is_git_repo() -> bool:
     path = Path(Path.cwd())
@@ -41,14 +45,17 @@ def generate_commit_message(diff: str) -> str:
             msg = client.models.generate_content(model="gemini-2.0-flash",
                                                  contents=query)
 
-            print(msg.text)
+            print(f"{GREEN}{msg.text}{RESET}")
             # generated_msg = "feat: Add poetry.lock and pyproject.toml files."
             # print(generated_msg)
             if continue_with_this_prompt(msg.text):
                 break
     except genai.errors.ClientError as error:
         if error.code == 400:
-            print(dict(error="Gemini API key not valid, check ~/.gmsg"))
+            print(
+                dict(
+                    error=f"{RED}Gemini API key not valid, check ~/.gmsg{RESET}"
+                ))
         else:
             print(error)
             sys.exit(1)
@@ -59,7 +66,7 @@ def continue_with_this_prompt(commit_msg: str) -> bool:
     if action in ("", "y", "Y"):
         print(f"Running: `git commit -m {commit_msg}`")
         print(
-            f"Message commited to git, to confirm you can run git commit --amend."
+            f"{GREEN}Message commited to git, to confirm you can run git commit --amend.{RESET}"
         )
         commit_message_to_git(commit_msg)
         return True

{gmsg-0.1.0.dist-info → gmsg-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: gmsg
-Version: 0.1.0
+Version: 0.1.1
 Summary: Generates git commit messages for you using AI.
 License: MIT
 Author: shiyan99s@gmail.com

gmsg-0.1.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+gmsg/__init__.py,sha256=9B4ZkbOBiBYYqv3Kh-mdlNn-yDPvkpXOqrm52Ng8CTk,23
+gmsg/api_key.py,sha256=92X--Gmwki2QxBSpMwo5xTl6I1g6GMw8FbCwVx8ojME,454
+gmsg/gmsg.py,sha256=hflPSahpCDgaCsFE33udZSkZ4hgB43UkCZZ9tcigTd4,3172
+gmsg-0.1.1.dist-info/METADATA,sha256=PWwm7v5MgB32x9ZCOXXw2ko39yK7Xk2uhYQyI4PK65w,458
+gmsg-0.1.1.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+gmsg-0.1.1.dist-info/entry_points.txt,sha256=LqLzIXT9IVenl_xEptCn4LqWHm34ypTaVMhoW5tFBSg,39
+gmsg-0.1.1.dist-info/RECORD,,

gmsg-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-gmsg/__init__.py,sha256=9B4ZkbOBiBYYqv3Kh-mdlNn-yDPvkpXOqrm52Ng8CTk,23
-gmsg/api_key.py,sha256=92X--Gmwki2QxBSpMwo5xTl6I1g6GMw8FbCwVx8ojME,454
-gmsg/gmsg.py,sha256=NmQd55li6D5g8vR2hY1gxKXarm0XPSG9jyPO63RvS5g,3016
-gmsg-0.1.0.dist-info/METADATA,sha256=GusjYVbsPUYN-LV7Mn7lg30VvidTTJjFpmUdjfZCS94,458
-gmsg-0.1.0.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
-gmsg-0.1.0.dist-info/entry_points.txt,sha256=LqLzIXT9IVenl_xEptCn4LqWHm34ypTaVMhoW5tFBSg,39
-gmsg-0.1.0.dist-info/RECORD,,