Update app.py

app.py

@@ -1,15 +1,11 @@
-import os
 import gradio as gr
 import requests
-
-# Retrieve the API key from the environment variable
-api_key = os.getenv('CODEPAL_API_KEY')
-if not api_key:
-    raise ValueError("API key not found. Please set the CODEPAL_API_KEY environment variable.")
+import os
 
 # Function to call CodePal's Code Generator API
 def generate_code(instructions, language, generation_type):
     api_url = 'https://api.codepal.ai/v1/code-generator/query'
+    api_key = os.getenv('CODEPAL_API_KEY')
     headers = {'Authorization': f'Bearer {api_key}'}
     data = {
         'instructions': instructions,
@@ -29,6 +25,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# Code Generator Interface\nGenerate code snippets using CodePal's Code Generator API.")
     with gr.Row():
         with gr.Column():
+            login_button = gr.LoginButton()
             instructions_input = gr.Textbox(label='Function Description', placeholder='Describe the function to generate...')
             language_input = gr.Dropdown(label='Programming Language', choices=['python', 'javascript', 'go', 'java', 'csharp'])
             generation_type_input = gr.Radio(label='Generation Type', choices=['minimal', 'standard', 'documented'], value='standard')