Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from browser_use import Agent
 from langchain_openai import ChatOpenAI
 
-# 1. Функция для установки браузера (нужна для Gradio Spaces)
+# 1. Функция для установки браузера
 async def install_playwright():
     process = await asyncio.create_subprocess_exec(
         'playwright', 'install', 'chromium',
@@ -15,20 +15,22 @@ async def install_playwright():
 
 # 2. Основная функция агента
 async def run_browser_agent(task):
-    # ПЫТАЕМСЯ ВЗЯТЬ КЛЮЧ ИЗ СЕКРЕТОВ (Settings -> Variables and secrets)
     api_key = os.getenv("NVIDIA_API_KEY")
     
     if not api_key:
-        return "ОШИБКА: Секрет 'NVIDIA_API_KEY' не найден в настройках Space!"
+        return "ОШИБКА: Секрет 'NVIDIA_API_KEY' не найден в Settings -> Secrets!"
 
     try:
-        # Используем ChatOpenAI интерфейс для работы с NVIDIA
+        # Создаем модель
         llm = ChatOpenAI(
             model="meta/llama-3.1-405b-instruct",
             openai_api_key=api_key,
             base_url="https://integrate.api.nvidia.com/v1"
         )
         
+        # ХАК: Вручную добавляем атрибут, который требует browser-use
+        llm.provider = "openai" 
+
         agent = Agent(
             task=task,
             llm=llm,
@@ -38,22 +40,24 @@ async def run_browser_agent(task):
         )
         
         result = await agent.run()
+        # Возвращаем финальный результат как строку
         return str(result.final_result())
+        
     except Exception as e:
         return f"Произошла ошибка: {str(e)}"
 
-# Устанавливаем браузер при запуске
-asyncio.run(install_playwright())
+# Установка браузера при старте (один раз)
+try:
+    asyncio.run(install_playwright())
+except:
+    pass
 
-# 3. Интерфейс Gradio (теперь только одно поле для задания)
+# 3. Интерфейс Gradio
 demo = gr.Interface(
     fn=lambda task: asyncio.run(run_browser_agent(task)),
-    inputs=[
-        gr.Textbox(label="Что должен сделать агент?", placeholder="Например: Найди цену золота на сегодня")
-    ],
+    inputs=[gr.Textbox(label="Задание", placeholder="Например: какая погода в Киеве?")],
     outputs=gr.Textbox(label="Результат"),
-    title="Browser-Use AI (NVIDIA Auto-Key)",
-    description="Агент берет ключ из Secrets и использует NVIDIA API."
+    title="Browser-Use AI (Fixed Version)"
 )
 
 if __name__ == "__main__":