fl example working with 3 hospitals

Author: wip-abramson

projects/aries-fl/Dockerfile.hospitalcontroller

@@ -0,0 +1,38 @@
+FROM continuumio/miniconda3
+
+ENV WORKSPACE /workspace
+
+ARG jupyter_port
+ENV env_jupyter_port=jupyter_port
+
+# Setup workspace environment
+RUN apt-get update && apt-get install -y gcc
+# RUN conda install jupyter notebook=5.7.8
+RUN conda install jupyter notebook=6.1.5
+
+
+ADD projects/aries-fl/requirements.txt .
+
+#
+# .
+RUN pip install -r requirements.txt
+
+#RUN pip install aiohttp
+#RUN export PYTHONPATH="$PYTHONPATH:/aries_basic_controller"
+#ENV PYTHONPATH "${PYTHONPATH}:/aries_basic_controller"
+#RUN export JUPYTER_PATH="$JUPYTER_PATH:/aries_basic_controller"
+# Create jupyter notebook workspace
+#RUN mkdir $WORKSPACE
+WORKDIR $WORKSPACE
+
+
+# Make the image start the jupyer notebook
+COPY scripts/jupyter-entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
+
+
+
+
+
+# ENTRYPOINT ["/entrypoint.sh", "$env_jupyter_port"]
+ENTRYPOINT ["/entrypoint.sh", "8888"]

projects/aries-fl/Dockerfile.ppmlcontroller

@@ -25,6 +25,7 @@ RUN pip install -r requirements.txt
 #RUN mkdir $WORKSPACE
 WORKDIR $WORKSPACE
 
+
 # Make the image start the jupyer notebook
 COPY scripts/jupyter-entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh

projects/aries-fl/docker-compose.yml

@@ -176,7 +176,8 @@ services:
     networks:
       - indy_demo
     volumes:
-      - ./notebooks/hospital:/workspace
+      - ./notebooks/hospital1:/workspace
+      - ./hospital:/workspace/hospital
     ports:
       - "${HOSPITAL1_JUPYTER_PORT}:8888"
       - ${HOSPITAL1_WEBHOOK_PORT}:${HOSPITAL1_WEBHOOK_PORT}
@@ -221,7 +222,8 @@ services:
     networks:
       - indy_demo
     volumes:
-      - ./notebooks/hospital:/workspace
+      - ./notebooks/hospital2:/workspace
+      - ./hospital:/workspace/hospital
     ports:
       - "${HOSPITAL2_JUPYTER_PORT}:8888"
       - ${HOSPITAL2_WEBHOOK_PORT}:${HOSPITAL2_WEBHOOK_PORT}
@@ -266,7 +268,8 @@ services:
     networks:
       - indy_demo
     volumes:
-      - ./notebooks/hospital:/workspace
+      - ./notebooks/hospital3:/workspace
+      - ./hospital:/workspace/hospital
     ports:
       - "${HOSPITAL3_JUPYTER_PORT}:8888"
       - ${HOSPITAL3_WEBHOOK_PORT}:${HOSPITAL3_WEBHOOK_PORT}

projects/aries-fl/hospital/__init__.py

@@ -0,0 +1 @@
+from hospital.hospital import Hospital

projects/aries-fl/notebooks/hospital/hospital.py renamed to projects/aries-fl/hospital/hospital.py

@@ -45,12 +45,12 @@ def __init__(self, agent_config, data_location):
     def _register_agent_listeners(self):
 
         loop = asyncio.get_event_loop()
-        loop.create_task(self.agent_controller.listen_webhooks())
+        loop.run_until_complete(self.agent_controller.listen_webhooks())
 
         listeners = [{
-            "handler": self._ml_messages_handler,
-            "topic": "basicmessages"
-        },
+                "handler": self._ml_messages_handler,
+                "topic": "basicmessages"
+            },
             {
                 "handler": self._connection_handler,
                 "topic": "connections"
@@ -128,6 +128,7 @@ def _connection_handler(self, payload):
                         response = loop.run_until_complete(self.agent_controller.proofs.send_request(proof_request_web_request))
                     else:
                         # No Auth Policy set
+                        print("No Auth Policy set")
                         connection["is_trusted"].set_result(True)
                     break
 
@@ -288,7 +289,9 @@ def _process_data(self, file_path):
         for data in y_train.values:
             y_train_data.append([data])
         y_train_data = torch.tensor(y_train_data).float()
-
+        
+        print("Data Processed")
+        
         return {
             "x": x_train_data,
             "y": y_train_data
@@ -367,7 +370,7 @@ def establish_research_connection(self, invitation):
         }
 
         self.pending_connections.append(pending_connection)
-
+        print("Establishing connection")
         loop.run_until_complete(pending_connection["is_trusted"])
 
         if pending_connection["is_trusted"].result():

projects/aries-fl/manage

@@ -174,8 +174,8 @@ configureEnvironment() {
   export HOSPITAL2_ADMIN_PORT=8061
   export HOSPITAL2_HTTP_PORT=8060
   export HOSPITAL2_WEBHOOK_PORT=8062
-  export HOSPITAL2_WEBHOOK_URL=${HOSPITAL1_WEBHOOK_URL:-http://$DOCKERHOST:$HOSPITAL2_WEBHOOK_PORT}
-  export HOSPITAL2_AGENT_ENDPOINT=${HOSPITAL1_AGENT_ENDPOINT:-http://$DOCKERHOST:$HOSPITAL2_HTTP_PORT}
+  export HOSPITAL2_WEBHOOK_URL=${HOSPITAL2_WEBHOOK_URL:-http://$DOCKERHOST:$HOSPITAL2_WEBHOOK_PORT}
+  export HOSPITAL2_AGENT_ENDPOINT=${HOSPITAL2_AGENT_ENDPOINT:-http://$DOCKERHOST:$HOSPITAL2_HTTP_PORT}
   export HOSPITAL2_WALLET_SEED="hospital2_seed_00000000000000000"
   export HOSPITAL2_JUPYTER_PORT="8892"
   export HOSPITAL2_WALLET_KEY="hospital2_key_00000000000000000000000"