import sys
import openai
import json

def main(user_message):
    # Set your OpenAI API key from the environment, or replace with your key here
    openai.api_key = "sk-proj-CMVUSsmXIr-Da3a8bpAByG0v2FD1hxEahGs7CqTz7tcegAWGP03OitAAD5LQVf_z5ZUucDWZ10pSHXJVzoWZeGCHueskkC5IMLccUldlvTlsfUA"
    
    try:
        # Call the OpenAI ChatCompletion API
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": user_message}],
            assistant_id="asst_0LyJEWQS90Sx0oEIYBVWE47C"  # Assistant ID
        )

        assistant_message = response['choices'][0]['message']['content']
        print(json.dumps({"response": assistant_message}))

    except Exception as e:
        print(json.dumps({"response": f"An error occurred: {str(e)}"}))

if __name__ == "__main__":
    if len(sys.argv) > 1:
        user_message = sys.argv[1]
        main(user_message)
    else:
        print(json.dumps({"response": "No message provided"}))