diff --git a/README.md b/README.md index 1a4ae19..e197c8f 100644 --- a/README.md +++ b/README.md @@ -56,6 +56,16 @@ prompt = ''' result, notes = chat(prompt, test=False) print(result) ``` +## Defining Api Keys ( .env file ) + +```bash + # .env file content + + OPENAI_API_KEY = "sk-proj-ABCDEFGIJKLMNOPQRSTUXVZ" + EVDS_API_KEY=ABCDEFGIJKLMNOP + +``` + ## New in this version diff --git a/evdschat/core/chat.py b/evdschat/core/chat.py index f8f352e..3183c5a 100644 --- a/evdschat/core/chat.py +++ b/evdschat/core/chat.py @@ -87,7 +87,11 @@ def chat_console(): result, notes = chat(prompt=args.prompt, debug=args.debug, test=args.test) print(result) print(notes) - result.to_excel(f"{args.file_name}.xlsx") + def correct(x:str): + if x.endswith('.xlsx') : return x + return f'{x}.xlsx' + + result.to_excel(f"{correct(args.file_name)}") if __name__ == "__main__":