import openaiimport gradio as grfrom datetime import datetimeimport osfrom dotenv import load_dotenv
load_dotenv()openai.api_key = os.getenv("OPENAI_API_KEY")
def ask_openai(question):    try:        response = openai.ChatCompletion.create(            model="gpt-4",            messages=[                {"role": "system", "content": "You are a funky and friendly assistant named ZapBot."},                {"role": "user", "content": question}            ],            temperature=0.3,            max_tokens=200        )        return response['choices'][0]['message']['content']    except Exception as e:        return f"Error: {str(e)}"
custom_css = """body { background-color: #121212; color: #f0f0f0; }h1, p { color: #00FFB3 !important; }.gr-textbox textarea { height: 100px !important; font-size: 16px; }"""
with gr.Blocks(css=custom_css) as demo:    gr.HTML("""    <h1>ZapBot: The Chat With a Spark!</h1>    <p>Always ready to zap some answers your way</p>    """)    chatbot = gr.Chatbot(height=450)    message = gr.Textbox(placeholder="Type your question here...", lines=4)    send_btn = gr.Button("Send", size="lg")
    def respond(user_message, chat_history):        if user_message.strip() == "":            return "", chat_history        bot_response = ask_openai(user_message)        timestamp = datetime.now().strftime("%H:%M")        chat_history.append((f"({timestamp})\n{user_message}", f"({timestamp})\n{bot_response}"))        return "", chat_history
    send_btn.click(respond, inputs=[message, chatbot], outputs=[message, chatbot])    message.submit(respond, inputs=[message, chatbot], outputs=[message, chatbot])
demo.launch(share=True)