Update app.py

app.py

@@ -5,6 +5,10 @@ import os
 import gradio as gr
 from gradio import ChatMessage
 import time
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 #load_dotenv()
 
@@ -17,9 +21,9 @@ import time
 api_key = os.getenv('GEMINI_API_KEY')
 
 if api_key:
-    print("API Key Loaded Successfully!")
+    logging.info("API Key Loaded Successfully!")
 else:
-    print("API Key Missing!")
+    logging.info("API Key Missing!")
 
 
 client = OpenAI(api_key=api_key,
@@ -80,9 +84,9 @@ def llm_response(message, history):
             messages= history)
 
         parsed_response = json.loads(llm_response.choices[0].message.content)
-        print("________________________________")
-        print(parsed_response)
-        print("________________________________")
+        logging.info("________________________________")
+        logging.info(parsed_response)
+        logging.info("________________________________")
         time.sleep(2)
         history.append({"role": "assistant", "content": parsed_response['content']})
         if parsed_response['step'] == 'result':