Create components/api_manager.py

components/api_manager.py

@@ -0,0 +1,72 @@
+import os
+import requests
+
+class APIManager:
+    def __init__(self):
+        self.minds_api_key = os.getenv('MINDSDB_API_KEY')
+        self.supabase_password = os.getenv('SUPABASE_PASSWORD')
+        self.minds_name = 'precision_medicine_mind'
+        self.headers = {
+            'Authorization': f'Bearer {self.minds_api_key}',
+            'Content-Type': 'application/json'
+        }
+
+    def create_mind(self):
+        supabase_config = {
+            "description": "Your Supabase Database",
+            "type": "postgres",
+            "connection_args": {
+                "user": "postgres.mgvepfrmsojwkoudycez",
+                "password": self.supabase_password,
+                "host": "aws-0-ap-southeast-1.pooler.supabase.com",
+                "port": "6543",
+                "database": "postgres"
+            },
+            "tables": ["Precision_medicine"]
+        }
+
+        payload = {
+            "name": self.minds_name,
+            "data_source_configs": [supabase_config]
+        }
+
+        try:
+            response = requests.post(
+                url='https://llm.mdb.ai/minds',
+                json=payload,
+                headers=self.headers
+            )
+            
+            if response.status_code == 201:
+                return f"Mind '{self.minds_name}' created successfully!"
+            else:
+                return "Mind already exists"
+
+        except requests.exceptions.HTTPError as http_err:
+            return f"HTTP error occurred: {http_err}"
+        except Exception as e:
+            return f"An error occurred: {e}"
+
+    def query_mind(self, query):
+        query_payload = {
+            "model": self.minds_name,
+            "messages": [
+                {"role": "user", "content": query}
+            ]
+        }
+
+        try:
+            query_response = requests.post(
+                url='https://llm.mdb.ai/chat/completions',
+                json=query_payload,
+                headers=self.headers
+            )
+
+            if query_response.status_code == 200:
+                response_data = query_response.json()
+                return response_data['choices'][0]['message']['content']
+            else:
+                return f"Error querying the Mind: {query_response.status_code} - {query_response.text}"
+
+        except Exception as e:
+            return f"Error querying the Mind: {e}"