Update app.py

app.py

@@ -11,7 +11,8 @@ from packaging import version
 from PIL import Image, ImageDraw
 import functools
 import emoji
-from langchain.llms.openai import OpenAI
+from langchain_community.chat_models import ChatOpenAI
+from langchain.schema import HumanMessage
 from caption_anything.model import CaptionAnything
 from caption_anything.utils.image_editing_utils import create_bubble_frame
 from caption_anything.utils.utils import mask_painter, seg_model_map, prepare_segmenter, image_resize
@@ -513,8 +514,9 @@ def validate_api_key(api_key):
     api_key = str(api_key).strip()
     print(api_key)
     try:
-        test_llm = OpenAI(model_name="gpt-3.5-turbo", temperature=0, openai_api_key=api_key)
-        response = test_llm("Test API call")
+        test_llm = ChatOpenAI(model_name="gpt-4o", temperature=0, openai_api_key=api_key)
+        print("test_llm")
+        response = test_llm([HumanMessage(content='Hello')])
         print(response)
         return True
     except Exception as e:
@@ -530,9 +532,11 @@ def init_openai_api_key(api_key=""):
         print(api_key)
         if validate_api_key(api_key):
             try:
-                text_refiner = build_text_refiner(args.text_refiner, args.device, args, api_key)
-                assert len(text_refiner.llm('hi')) > 0 # test
-                visual_chatgpt = ConversationBot(shared_chatbot_tools, api_key)
+                # text_refiner = build_text_refiner(args.text_refiner, args.device, args, api_key)
+                # assert len(text_refiner.llm('hi')) > 0 # test
+                text_refiner = None
+                print("text refiner")
+                visual_chatgpt = ConversationBot(shared_chatbot_tools, api_key=api_key)
             except Exception as e:
                 print(f"Error initializing TextRefiner or ConversationBot: {e}")
                 text_refiner = None
@@ -543,7 +547,7 @@ def init_openai_api_key(api_key=""):
         print("API key is too short.")
     print(text_refiner)
     openai_available = text_refiner is not None
-    if openai_available:
+    if visual_chatgpt:
 
         global gpt_state
         gpt_state=1