Create gemini_tool.py

gemini_tool.py

@@ -0,0 +1,28 @@
+import os
+import google.generativeai as genai
+
+# Fetch the API key from the environment variable (Space secret)
+api_key = os.getenv("GOOGLE_API_KEY")
+
+if not api_key:
+    raise ValueError("GOOGLE_API_KEY secret not set in Hugging Face Space settings.")
+
+# Configure the Gemini API
+genai.configure(api_key=api_key)
+
+# Initialize the model
+# You can change 'gemini-1.5-flash-latest' to 'gemini-1.5-pro-latest' or other models
+model = genai.GenerativeModel('gemini-1.5-flash-latest')
+
+# Get user input from the terminal
+prompt = input("Enter your prompt for Gemini: ")
+
+print("\nGenerating response...")
+
+# Generate content
+response = model.generate_content(prompt)
+
+# Print the response text
+print("\n--- Gemini Response ---")
+print(response.text)
+print("-----------------------\n")