diff --git a/hass/custom_components/gemini_api/__init__.py b/hass/custom_components/gemini_api/__init__.py
index f0e456c..12048fe 100644
--- a/hass/custom_components/gemini_api/__init__.py
+++ b/hass/custom_components/gemini_api/__init__.py
@@ -1,10 +1,9 @@
 #imports
-import json
-import socket
-import ast
 import base64
+import requests
+import json
 
-from homeassistant.core import SupportsResponse
+from homeassistant.helpers import SupportsResponse
 
 DOMAIN = 'gemini_api'
 
@@ -52,6 +51,4 @@ def generate_text(call):
             "text": response.json()["candidates"][0]["content"]["parts"][0]["text"]
         }
 
-
     hass.services.register(DOMAIN, 'generate_text', generate_text, supports_response=SupportsResponse.ONLY)
-    return True