Update app.py

app.py

@@ -14,23 +14,37 @@ logger = logging.getLogger(__name__)
 MAX_TEXT_LENGTH = 5000
 TEMP_FILES = []
 
-# Инициализация клиентов через переменные окружения
-HF_API_KEY = os.environ.get("HF_API_KEY")
-if not HF_API_KEY:
-    raise ValueError("HF_API_KEY not found in environment variables")
-
-image_client = InferenceClient(token=HF_API_KEY)
-chat_client = InferenceClient(provider="together", api_key=HF_API_KEY)
+# Инициализация клиентов (будет выполнена после получения API ключа)
+image_client = None
+chat_client = None
 
 # Стиль интерфейса
 theme = gr.themes.Soft()
 
+def initialize_clients(api_key):
+    """Инициализирует клиенты API с полученным ключом"""
+    global image_client, chat_client
+    try:
+        image_client = InferenceClient(token=api_key)
+        chat_client = InferenceClient(provider="together", api_key=api_key)
+        return True
+    except Exception as e:
+        logger.error(f"Initialization error: {str(e)}")
+        return False
+
 # Парсер промптов
 def parse_image_prompts(text):
     return re.findall(r'<image>(.*?)<\/image>', text, re.DOTALL)
 
 # Генератор изображений
-def generate_image(prompt):
+def generate_image(prompt, api_key):
+    """Генерирует изображение с проверкой API ключа"""
+    if not api_key:
+        return None, "⚠️ Please enter your API key first"
+    
+    if not initialize_clients(api_key):
+        return None, "⚠️ Invalid API key"
+    
     try:
         response = image_client.text_to_image(
             prompt.strip(),
@@ -51,7 +65,14 @@ def generate_image(prompt):
         return None, f"⚠️ Error: {str(e)}"
 
 # Генератор текста
-async def generate_text(input_text):
+async def generate_text(input_text, api_key):
+    """Генерирует текст с проверкой API ключа"""
+    if not api_key:
+        return None, "⚠️ Please enter your API key first"
+    
+    if not initialize_clients(api_key):
+        return None, "⚠️ Invalid API key"
+    
     try:
         completion = chat_client.chat.completions.create(
             model="deepseek-ai/DeepSeek-R1",
@@ -65,13 +86,23 @@ async def generate_text(input_text):
 
 # Интерфейс
 def create_interface():
-    with gr.Blocks(theme=theme, title="Secure AI Studio") as demo:
-        gr.Markdown("# 🔒 Secure Content Generator")
+    with gr.Blocks(theme=theme, title="AI Studio") as demo:
+        gr.Markdown("# 🔑 AI Content Generator")
         
+        # Секция для ввода API ключа
+        with gr.Row():
+            api_key = gr.Textbox(
+                label="HuggingFace API Key",
+                type="password",
+                placeholder="Enter your HF_API_KEY here...",
+                info="Get your key from https://huggingface.co/settings/tokens"
+            )
+        
+        # Основная секция
         with gr.Row():
             with gr.Column():
                 input_text = gr.Textbox(
-                    label="Input",
+                    label="Input Text",
                     placeholder="Enter text with <image>prompt</image> tags...",
                     lines=5
                 )
@@ -81,19 +112,20 @@ def create_interface():
                     image_btn = gr.Button("Generate Images", variant="secondary")
                 
             with gr.Column():
-                output_text = gr.Textbox(label="Result", interactive=False)
-                gallery = gr.Gallery(label="Images", columns=2)
+                output_text = gr.Textbox(label="Generated Text", interactive=False)
+                gallery = gr.Gallery(label="Generated Images", columns=2)
                 status = gr.HTML()
 
+        # Обработчики событий
         text_btn.click(
             generate_text,
-            inputs=input_text,
+            inputs=[input_text, api_key],
             outputs=[output_text, status]
         )
         
         image_btn.click(
-            lambda x: [generate_image(p) for p in parse_image_prompts(x)],
-            inputs=output_text,
+            lambda text, key: [generate_image(p, key) for p in parse_image_prompts(text)],
+            inputs=[output_text, api_key],
             outputs=[gallery, status]
         )
 
@@ -103,8 +135,13 @@ def create_interface():
 if __name__ == "__main__":
     try:
         demo = create_interface()
-        demo.launch(server_port=7860)
+        demo.launch(
+            server_port=7860,
+            show_error=True,
+            share=False
+        )
     finally:
+        # Очистка временных файлов
         for file in TEMP_FILES:
             try:
                 Path(file).unlink(missing_ok=True)