thanhnt-cf commited on
Commit
44e85d0
·
1 Parent(s): ed3afa6

run dotenv even on huggingface space

Browse files
app.py CHANGED
@@ -1,13 +1,13 @@
1
  import json
2
  import gradio as gr
3
- # from dotenv import load_dotenv
4
  from llm_api.exceptions import RefusalError
5
  # from llm_api.langchain_api import extract_info, follow_structure
6
  from llm_api.native_api import extract_info, follow_structure
7
  from openai import BadRequestError
8
  # from utils import log_input, log_output
9
 
10
- # load_dotenv('.env', override=True)
11
 
12
  def extract_free_fn(gallery, provider, schema):
13
  # write schema to file before calling extract_free_properties
 
1
  import json
2
  import gradio as gr
3
+ from dotenv import load_dotenv
4
  from llm_api.exceptions import RefusalError
5
  # from llm_api.langchain_api import extract_info, follow_structure
6
  from llm_api.native_api import extract_info, follow_structure
7
  from openai import BadRequestError
8
  # from utils import log_input, log_output
9
 
10
+ load_dotenv('.env', override=True)
11
 
12
  def extract_free_fn(gallery, provider, schema):
13
  # write schema to file before calling extract_free_properties
llm_api/anthropic_api.py CHANGED
@@ -1,9 +1,9 @@
1
  from anthropic import Anthropic
2
- # from dotenv import load_dotenv
3
  from llm_api.utils import get_data_format, get_image_data
4
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
5
 
6
- # load_dotenv(override=True)
7
  client = Anthropic()
8
 
9
  def extract_info(img_paths, schema):
 
1
  from anthropic import Anthropic
2
+ from dotenv import load_dotenv
3
  from llm_api.utils import get_data_format, get_image_data
4
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
5
 
6
+ load_dotenv(override=True)
7
  client = Anthropic()
8
 
9
  def extract_info(img_paths, schema):
llm_api/langchain_api.py CHANGED
@@ -1,13 +1,13 @@
1
  from .utils import get_data_format, get_image_data
2
  from langchain_anthropic import ChatAnthropic
3
  from langchain_openai import ChatOpenAI
4
- # from dotenv import load_dotenv
5
  from langchain_core.messages import HumanMessage, SystemMessage
6
  from schema_structured import Garment as StructuredGarment
7
  from schema_free import Garment as FreeGarment
8
 
9
 
10
- # load_dotenv()
11
 
12
 
13
  anthropic_llm = ChatAnthropic(
 
1
  from .utils import get_data_format, get_image_data
2
  from langchain_anthropic import ChatAnthropic
3
  from langchain_openai import ChatOpenAI
4
+ from dotenv import load_dotenv
5
  from langchain_core.messages import HumanMessage, SystemMessage
6
  from schema_structured import Garment as StructuredGarment
7
  from schema_free import Garment as FreeGarment
8
 
9
 
10
+ load_dotenv()
11
 
12
 
13
  anthropic_llm = ChatAnthropic(
llm_api/native_api.py CHANGED
@@ -1,5 +1,4 @@
1
  import base64
2
- # from dotenv import load_dotenv
3
  from . import openai_api as openai_info_extractor
4
  from . import anthropic_api as anthropic_extractor
5
 
 
1
  import base64
 
2
  from . import openai_api as openai_info_extractor
3
  from . import anthropic_api as anthropic_extractor
4
 
llm_api/openai_api.py CHANGED
@@ -1,11 +1,11 @@
1
  import json
2
- # from dotenv import load_dotenv
3
  from .utils import get_data_format, get_image_data
4
  from openai import OpenAI, BadRequestError
5
  from .exceptions import RefusalError
6
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
7
 
8
- # load_dotenv(override=True)
9
  client = OpenAI()
10
 
11
 
 
1
  import json
2
+ from dotenv import load_dotenv
3
  from .utils import get_data_format, get_image_data
4
  from openai import OpenAI, BadRequestError
5
  from .exceptions import RefusalError
6
  from .constants import EXTRACT_INFO_HUMAN_MESSAGE, EXTRACT_INFO_SYSTEM_MESSAGE,FOLLOW_SCHEMA_HUMAN_MESSAGE, FOLLOW_SCHEMA_SYSTEM_MESSAGE
7
 
8
+ load_dotenv(override=True)
9
  client = OpenAI()
10
 
11
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  openai
2
  anthropic
3
- pydantic
 
 
1
  openai
2
  anthropic
3
+ pydantic
4
+ python-dotenv