Update app.py

app.py

@@ -1,45 +1,58 @@
 import gradio as gr
 from gradio_client import Client
+import shutil
 import os
 
-# Используем API существующего мощного спейса (например, официальный Shap-E от OpenAI или аналоги)
-# Мы подключаемся к нему как клиент. Это БЕСПЛАТНО.
+# Подключаемся к стабильному API Shap-E
+# Если этот адрес перестанет работать, его можно заменить на другой активный спейс
+API_URL = "huggingface-projects/Shap-E"
+
 def generate_3d(prompt):
+    if not prompt:
+        return None, "Введите описание!"
+    
     try:
-        # Подключаемся к рабочему спейсу с Shap-E
-        # Примечание: если этот спейс уснет, можно найти другой активный с такой же моделью
-        client = Client("huggingface-projects/Shap-E") 
-        
-        # Вызываем функцию генерации у стороннего спейса
-        # Обычно это функция .predict()
-        result = client.predict(
-            prompt,	# Твой текст
+        client = Client(API_URL)
+        # Вызываем генерацию
+        result_path = client.predict(
+            prompt=prompt,
             api_name="/predict"
         )
         
-        # Результатом будет путь к файлу .obj или .glb
-        return result
+        # result_path — это путь к временному файлу (например, .glb)
+        # Создаем копию в нашей рабочей директории, чтобы Gradio мог его отдать
+        final_filename = f"model_{os.path.basename(result_path)}"
+        shutil.copy(result_path, final_filename)
+        
+        return final_filename, final_filename
     except Exception as e:
-        return f"Ошибка связи с сервером 3D: {str(e)}"
+        return None, f"Ошибка: {str(e)}. Попробуйте другой промпт или подождите."
 
 css = """
 .gradio-container { background-color: #050505 !important; color: #d8b4fe !important; }
-.main-btn { background: linear-gradient(to right, #4c1d95, #2563eb) !important; border: none !important; }
+.main-btn { background: linear-gradient(to right, #4c1d95, #2563eb) !important; border: none !important; color: white !important; }
 """
 
-with gr.Blocks(css=css) as demo:
-    gr.Markdown("# 🚀 Free 3D Forge (No GPU Required)")
-    gr.Markdown("Этот инструмент использует внешние мощности для генерации 3D. Ты не платишь ни копейки!")
+with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
+    gr.Markdown("# 🚀 3D Forge: Download Edition")
+    gr.Markdown("Генерация 3D моделей через внешнее API. Бесплатно и доступно для скачивания.")
     
     with gr.Row():
-        with gr.Column():
-            inp = gr.Textbox(label="Описание объекта", placeholder="A red dragon head, low poly")
-            btn = gr.Button("⚒️ СОЗДАТЬ 3D", variant="primary", elem_classes="main-btn")
+        with gr.Column(scale=1):
+            inp = gr.Textbox(label="Что создаем?", placeholder="A futuristic cyber-helmet, neon details")
+            btn = gr.Button("⚒️ СГЕНЕРИРОВАТЬ", variant="primary", elem_classes="main-btn")
+            error_msg = gr.Markdown("") # Для вывода ошибок
         
-        with gr.Column():
-            # Model3D — крутой компонент Gradio, который позволяет крутить модель мышкой!
-            out = gr.Model3D(label="3D Preview (можно крутить)")
+        with gr.Column(scale=2):
+            # Предпросмотр модели (можно крутить)
+            out_preview = gr.Model3D(label="3D Preview")
+            # Кнопка для скачивания самого файла
+            out_file = gr.File(label="📥 СКАЧАТЬ МОДЕЛЬ")
 
-    btn.click(fn=generate_3d, inputs=inp, outputs=out)
+    btn.click(
+        fn=generate_3d, 
+        inputs=inp, 
+        outputs=[out_preview, out_file]
+    )
 
 demo.launch()