Update app.py

app.py

@@ -32,8 +32,7 @@ def initialize_clients(api_key):
         # Получение списка голосов для TTS
         asyncio.run(load_tts_voices())
         return True
-    except Exception as e:
-        logger.error(f"Initialization error: {str(e)}")
+    except Exception:
         return False
 
 async def load_tts_voices():
@@ -47,12 +46,9 @@ def clean_text(text):
     return text.strip()
 
 # Генерация аудио
-async def generate_audio(text, voice, api_key):
-    if not api_key:
-        return None, "⚠️ Please enter API key first"
-    
+async def generate_audio(text, voice):
     if not text.strip():
-        return None, "⚠️ Please enter text"
+        return None
     
     try:
         text = clean_text(text)
@@ -62,24 +58,58 @@ async def generate_audio(text, voice, api_key):
             tmp_path = tmp_file.name
             await communicate.save(tmp_path)
             TEMP_FILES.append(tmp_path)
-            return tmp_path, None
+            return tmp_path
             
-    except Exception as e:
-        logger.error(f"TTS error: {str(e)}")
-        return None, f"⚠️ Audio error: {str(e)}"
+    except Exception:
+        return None
 
-# Остальные функции (generate_image, generate_text) остаются без изменений...
+# Генерация изображений
+def generate_image(prompt):
+    if not prompt.strip():
+        return None
+    
+    try:
+        response = image_client.text_to_image(
+            prompt.strip(),
+            model="nerijs/dark-fantasy-illustration-flux",
+            negative_prompt="text, watermark, low quality",
+            guidance_scale=9,
+            height=512,
+            width=512
+        )
+        
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+            tmp_file.write(response)
+            TEMP_FILES.append(tmp_file.name)
+            return tmp_file.name
+            
+    except Exception:
+        return None
 
+# Генерация текста
+async def generate_text(input_text):
+    if not input_text.strip():
+        return None
+    
+    try:
+        completion = chat_client.chat.completions.create(
+            model="deepseek-ai/DeepSeek-R1",
+            messages=[{"role": "user", "content": input_text}],
+            max_tokens=500
+        )
+        return completion.choices[0].message.content
+    except Exception:
+        return None
+
+# Интерфейс
 def create_interface():
     with gr.Blocks(theme=theme, title="AI Studio Pro") as demo:
         gr.Markdown("# 🎙️ Full AI Studio")
         
-        # Секция API ключа
+        # Секция API ключа (скрытая)
         api_key = gr.Textbox(
-            label="HuggingFace API Key",
-            type="password",
-            placeholder="Enter HF_API_KEY...",
-            info="Get key from https://huggingface.co/settings/tokens"
+            value=os.environ.get("HF_API_KEY", ""),
+            visible=False
         )
         
         with gr.Tabs():
@@ -91,23 +121,39 @@ def create_interface():
                         tts_voice = gr.Dropdown(
                             label="Voice",
                             choices=tts_voices,
-                            interactive=True
+                            value="en-US-ChristopherNeural (Male)" if tts_voices else None
                         )
                         tts_btn = gr.Button("Generate Speech", variant="primary")
                     
                     with gr.Column():
                         tts_audio = gr.Audio(label="Generated Audio")
-                        tts_status = gr.HTML()
                 
                 tts_btn.click(
                     generate_audio,
-                    [tts_text, tts_voice, api_key],
-                    [tts_audio, tts_status]
+                    [tts_text, tts_voice],
+                    tts_audio
                 )
 
             # Вкладка генерации текста и изображений
             with gr.Tab("🖼️ Text & Image Generation"):
-                # ... предыдущий код вкладки ...
+                with gr.Row():
+                    with gr.Column():
+                        input_text = gr.Textbox(
+                            label="Input Text",
+                            placeholder="Enter text with <image>prompt</image> tags...",
+                            lines=5
+                        )
+                        gen_btn = gr.Button("Generate Content", variant="primary")
+                    
+                    with gr.Column():
+                        output_text = gr.Textbox(label="Generated Text", interactive=False)
+                        gallery = gr.Gallery(label="Generated Images", columns=2)
+                
+                gen_btn.click(
+                    fn=lambda x: [generate_text(x)] + [generate_image(p) for p in parse_image_prompts(x)],
+                    inputs=input_text,
+                    outputs=[output_text, gallery]
+                )
 
         return demo
 
@@ -120,5 +166,5 @@ if __name__ == "__main__":
         for file in TEMP_FILES:
             try:
                 Path(file).unlink(missing_ok=True)
-            except Exception as e:
-                logger.warning(f"Cleanup error: {str(e)}")
+            except Exception:
+                pass