Update app.py

app.py

@@ -2,11 +2,9 @@ import google.generativeai as genai
 import gradio as gr
 
 
-def flip_text(prompt):
+def flip_text(prompt, api_key):
     def init_api_keys():
         try:
-            
-            api_key = 'AIzaSyBYxPTwLyP7D6Hiyydk3olQ1FeXoP78wmc'
             genai.configure(api_key=api_key)
         except Exception as exception:
             print("Error in initializing API keys:", exception)
@@ -16,10 +14,10 @@ def flip_text(prompt):
     def generate_response(prompt: str):
         # Set up the model
         generation_config = {
-            "temperature": 0.1,
+            "temperature": 1,
             "top_p": 1,
             "top_k": 1,
-            "max_output_tokens": 2048,
+            "max_output_tokens": 8192,
         }
         
         model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest", generation_config=generation_config)
@@ -45,11 +43,13 @@ with gr.Blocks() as demo:
     with gr.Tab("Базовые настройки"):
         with gr.Row():
             prompt = gr.Textbox(placeholder="Введите описание...", show_label=True, label='Описание:', lines=3)
+        with gr.Row():
+            api_key = gr.Textbox(placeholder="Введите api_key...", show_label=True, label='Описание:', lines=3)
     
     with gr.Column():
         text_button = gr.Button("Сгенерировать текст", variant='primary', elem_id="generate")
     with gr.Column():
         image_output = gr.Textbox(lines=30, elem_id='image_output')
-        text_button.click(flip_text, inputs=[prompt], outputs=image_output, concurrency_limit=48)
+        text_button.click(flip_text, inputs=[prompt, api_key], outputs=image_output, concurrency_limit=48)
 
 demo.launch()