adding logger

MM-REACT/app.py

@@ -18,6 +18,12 @@ from langchain.llms import AzureOpenAI
 from langchain.utilities import ImunAPIWrapper, ImunMultiAPIWrapper
 from openai.error import AuthenticationError, InvalidRequestError, RateLimitError
 import argparse
+import logging
+from opencensus.ext.azure.log_exporter import AzureLogHandler
+import uuid
+
+logger = None
+
 
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 BUG_FOUND_MSG = "There is a bug in the application!"
@@ -29,17 +35,17 @@ MAX_TOKENS = 512
 AGRS = None
 #####################################
 
-# resets memory
-def reset_memory(history):
-    # global memory
-    # memory.clear()
-    print ("clearning memory, loading langchain...")
-    load_chain()
-    history = []
-    return history, history
+
+def get_logger():
+    global logger
+    if logger is None:
+        logger = logging.getLogger(__name__)
+        logger.addHandler(AzureLogHandler())
+    return logger
+
 
 # load chain
-def load_chain(history):
+def load_chain(history, log_state):
     global ARGS
 
     if ARGS.openAIModel == 'openAIGPT35':
@@ -196,10 +202,16 @@ def load_chain(history):
     ]
 
     chain = initialize_agent(tools, llm, agent="conversational-assistant", verbose=True, memory=memory, return_intermediate_steps=True, max_iterations=4)
+    log_state = log_state or ""
+    print ("log_state {}".format(log_state))
+    log_state = str(uuid.uuid1())
     print("langchain reloaded")
+    # eproperties = {'custom_dimensions': {'key_1': 'value_1', 'key_2': 'value_2'}} 
+    properties = {'custom_dimensions': {'session': log_state}}
+    get_logger().warning("langchain reloaded", extra=properties)    
     history = []
-    history.append(("Show me what you got!", "Hi Human, I am ready to serve!"))    
-    return history, history, chain, gr.Textbox.update(visible=True), gr.Button.update(visible=True), gr.UploadButton.update(visible=True)
+    history.append(("Show me what you got!", "Hi Human, Please upload an image to get started!"))    
+    return history, history, chain, log_state, gr.Textbox.update(visible=True), gr.Button.update(visible=True), gr.UploadButton.update(visible=True)
 
 
 # executes input typed by human
@@ -231,13 +243,19 @@ class ChatWrapper:
         self.lock = Lock()
 
     def __call__(
-            self, inp: str, history: Optional[Tuple[str, str]], chain: Optional[ConversationChain]
+            self, inp: str, history: Optional[Tuple[str, str]], chain: Optional[ConversationChain], log_state
     ):
+        
         """Execute the chat functionality."""
         self.lock.acquire()
         try:
             print("\n==== date/time: " + str(datetime.datetime.now()) + " ====")
             print("inp: " + inp)
+
+            properties = {'custom_dimensions': {'session': log_state}}
+            get_logger().warning("inp: " + inp, extra=properties)
+
+
             history = history or []
             # If chain is None, that is because no API key was provided.
             output = "Please paste your OpenAI key from openai.com to use this app. " + str(datetime.datetime.now())
@@ -262,15 +280,49 @@ class ChatWrapper:
             self.lock.release()
 
         print (history)
+        properties = {'custom_dimensions': {'session': log_state}}
+        if outputs is None:
+            outputs = ""
+        get_logger().warning(str(json.dumps(outputs)), extra=properties)
+
         return history, history, ""
+   
+def add_image_with_path(state, chain, imagepath, log_state):
+    global ARGS
+    state = state or []
+
+    url_input_for_chain = "http://0.0.0.0:{}/file={}".format(ARGS.port, imagepath)
+
+    outputs = run_chain(chain, url_input_for_chain)
+
+    ########################
+    # multi line response handling
+    outputs = process_chain_output(outputs)
+
+    for i, output in enumerate(outputs):
+        if i==0:
+            # state.append((f"![](/file={imagepath})", output))    
+            state.append(((imagepath,), output))
+        else:
+            state.append((None, output))
+    
+
+    print (state)
+    properties = {'custom_dimensions': {'session': log_state}}
+    get_logger().warning("url_input_for_chain: " + url_input_for_chain, extra=properties)
+    if outputs is None:
+        outputs = ""
+    get_logger().warning(str(json.dumps(outputs)), extra=properties)
+    return state, state
+
 
 # upload image 
-def add_image(state, chain, image):
+def add_image(state, chain, image, log_state):
     global ARGS
     state = state or []
 
     url_input_for_chain = "http://0.0.0.0:{}/file={}".format(ARGS.port, image.name)
-
+    
     outputs = run_chain(chain, url_input_for_chain)
 
     ########################
@@ -286,6 +338,11 @@ def add_image(state, chain, image):
     
 
     print (state)
+    properties = {'custom_dimensions': {'session': log_state}}
+    get_logger().warning("url_input_for_chain: " + url_input_for_chain, extra=properties)
+    if outputs is None:
+        outputs = ""
+    get_logger().warning(str(json.dumps(outputs)), extra=properties)
     return state, state
 
 # extract image url from response and process differently
@@ -337,8 +394,9 @@ def init_and_kick_off():
         llm_state = gr.State()
         history_state = gr.State()
         chain_state = gr.State()      
+        log_state = gr.State() 
         
-        reset_btn = gr.Button(value="!!!CLICK to wake up the AI!!!", variant="secondary", elem_id="resetbtn").style(full_width=True)
+        reset_btn = gr.Button(value="!!!CLICK to wake up MM-REACT!!!", variant="secondary", elem_id="resetbtn").style(full_width=True)
 
         with gr.Row():
             chatbot = gr.Chatbot(elem_id="chatbot").style(height=620)
@@ -353,16 +411,27 @@ def init_and_kick_off():
             with gr.Column(scale=0.10, min_width=0):
                 btn = gr.UploadButton("📁", file_types=["image"], visible=False).style(full_width=True)
         
-        message.submit(chat, inputs=[message, history_state, chain_state],
+        # with gr.Row():
+        #     with gr.Column():
+        #         example1Image = gr.Image("images/money.png", interactive=False).style(height=100, width=100)
+        #     with gr.Column():
+        #         example1ImagePath = gr.Text("images/money.png", interactive=False, visible=False)
+        #     with gr.Column():
+        #         example1ImageButton = gr.Button(value="Try it", variant="secondary").style(full_width=True)
+        # example1ImageButton.click(add_image_with_path, inputs=[history_state, chain_state, example1ImagePath],
+        #         outputs=[history_state, chatbot])
+
+
+        message.submit(chat, inputs=[message, history_state, chain_state, log_state],
                     outputs=[chatbot, history_state, message])
 
-        submit.click(chat, inputs=[message, history_state, chain_state],
+        submit.click(chat, inputs=[message, history_state, chain_state, log_state],
                     outputs=[chatbot, history_state, message])
 
-        btn.upload(add_image, inputs=[history_state, chain_state, btn], outputs=[history_state, chatbot])
+        btn.upload(add_image, inputs=[history_state, chain_state, btn, log_state], outputs=[history_state, chatbot])
         
         # load the chain    
-        reset_btn.click(load_chain, inputs=[history_state], outputs=[chatbot, history_state, chain_state, message, submit, btn])
+        reset_btn.click(load_chain, inputs=[history_state, log_state], outputs=[chatbot, history_state, chain_state, log_state, message, submit, btn])
         
     # launch the app
     block.launch(server_name="0.0.0.0", server_port = ARGS.port)

requirements.txt

@@ -1,3 +1,4 @@
+opencensus
 imagesize==1.4.1
 gradio==3.21.0
 openai==0.26.4