Update main.py

main.py

@@ -1,6 +1,12 @@
 import SpaceFace as sf
-class GreetingEndpoint:
+import GPTSimple as ai
+class AIEndpoint:
     def receivedMessage(self, message, user: sf.User, responder: sf.Responder):
-        responder.respond(f"Hello {user.name}, you told me \"{message}\"")
+        if hasattr(user, 'chat'):
+            
+            responder.respond(user.chat.ask(message))
+        else:
+            chat = ai.conversation(base_url="deepinfra", model="meta-llama/Llama-3.2-90B-Vision-Instruct")
+            responder.respond(user.chat.ask(message))
         responder.close()
-sf.start_server([GreetingEndpoint()])
+sf.start_server([AIEndpoint()])