1

app.py

@@ -7,10 +7,11 @@ from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 
 import os
 
-OPENAI_API_KEY=""
-OPENAI_API_BASE=""
+OPENAI_API_KEY=os.getenv("OPENAI_API_KEY")
+OPENAI_API_BASE=os.getenv("OPENAI_API_BASE")
+DEP_NAME=os.getenv("deployment_name")
 
-llm=AzureChatOpenAI(deployment_name="bitservice_chat_35",openai_api_base=OPENAI_API_BASE,openai_api_key=OPENAI_API_KEY,openai_api_version="2023-03-15-preview",model_name="gpt-3.5-turbo")
+llm=AzureChatOpenAI(deployment_name=DEP_NAME,openai_api_base=OPENAI_API_BASE,openai_api_key=OPENAI_API_KEY,openai_api_version="2023-03-15-preview",model_name="gpt-3.5-turbo")
 
 import torch
 from transformers import BlipProcessor, BlipForConditionalGeneration
@@ -32,10 +33,6 @@ def describeImage(image_url):
   outputs = model.generate(**inputs)
   return processor.decode(outputs[0], skip_special_tokens=True)
 
-# description = describeImage('https://img.phb123.com/uploads/allimg/151026/7-151026161P2.jpg')
-
-# description
-
 from langchain.tools import BaseTool
 
 class DescribeImageTool(BaseTool):
@@ -65,12 +62,3 @@ agent = initialize_agent(
     )
 )
 
-image_url = 'https://img.phb123.com/uploads/allimg/151026/7-151026161P2.jpg'
-agent(f"Describe the following image:\n{image_url}")
-
-agent(f"How tall this tower looks in the following image:\n{image_url}")
-
-image_url = 'https://img.phb123.com/uploads/allimg/151026/7-151026161P2.jpg'
-agent(f"Please describe the following image:\n{image_url}")
-
-agent.memory.buffer