init, chat works properly

main.py

@@ -0,0 +1,28 @@
+import os
+import argparse
+from dotenv import load_dotenv
+from google import genai
+from google.genai import types
+
+
+
+load_dotenv()
+api_key = os.environ.get("GEMINI_API_KEY")
+if not api_key:
+    raise RuntimeError("API Key not found")
+    
+client = genai.Client(api_key=api_key)
+parser = argparse.ArgumentParser(description="Chatbot")
+parser.add_argument("user_prompt", type=str, help="User prompt")
+args = parser.parse_args()
+messages = [types.Content(role="user", parts=[types.Part(text=args.user_prompt)])]
+
+response = client.models.generate_content(
+    model='gemini-2.5-flash', contents=messages
+)
+if not response.usage_metadata:
+    raise RuntimeError("Cannot get usage metadata")
+
+print(f"Prompt tokens: {response.usage_metadata.prompt_token_count}")
+print(f"Response tokens: {response.usage_metadata.candidates_token_count}")
+print(response.text)