modified processing to use queue

Dockerfile

@@ -13,4 +13,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
 
 COPY . .
 
+#CMD ["python", "Consumer.py"] won't work. reliability?
+CMD ["python", "Consumer.py"]
+
 CMD ["python", "app.py"]

app.py

@@ -24,9 +24,39 @@ class RequestModel(BaseModel):
 def read_root():
     return {"message": "Hello From CodeBuddyAI!"}
 
+######################
+# Producer
+import pika
+RABBITMQ_URL = os.environ["RABBITMQ_URL"]
+QUEUE_NAME = os.environ["QUEUE_NAME"] 
 
-@app.post("/call/")
-def call(request: RequestModel):
+def get_connection():
+    params = pika.URLParameters(RABBITMQ_URL)
+    return pika.BlockingConnection(params)
+
+connection = get_connection()
+def send_message(data: dict):
+    connection = get_connection()
+    channel = connection.channel()
+
+    channel.queue_declare(queue=QUEUE_NAME, durable=True)
+
+    message = str(data)
+
+    channel.basic_publish(
+        exchange='',
+        routing_key=QUEUE_NAME,
+        body=message,
+        properties=pika.BasicProperties(
+            delivery_mode=2,  # make message persistent
+        )
+    )
+
+    connection.close()
+
+    return {"status": "sent", "data": data}
+######################
+def old_call(request: RequestModel):
     # fill with last state 
     try:
         state = json.loads(request.last_state) 
@@ -46,6 +76,9 @@ def call(request: RequestModel):
     return {"Data": answer}
 
     
+@app.post("/call/")
+def call(request: RequestModel):
+    return send_message(request.model_dump())
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)