Update app.py

app.py

@@ -1,155 +1,116 @@
 import gradio as gr
 import requests
+import json
 
-# ===== URL API =====
-URLS = {
+# ====== API BASE URL ======
+NEIROST4R_BASE = {
+    "ai_v1": "https://fenst4r.life/api/ai",
+    "ai_v2": "https://fenst4r.life/api/ai_v2",
+    "ai_v3": "https://fenst4r.life/api/ai_v3",
+    "ai_v4": "http://fenst4r.life/api/ai_v4",
+    "ai_v5": "http://fenst4r.life/api/ai_v5"
+}
+
+NEBLORA_BASE = {
     "chatgpt": "http://neblora.fenst4r.life:5400/gpt",
     "llama": "http://neblora.fenst4r.life:5500/llama",
-    "analyze_image": "http://neblora.fenst4r.life:4848/analyze-image",
     "draw": "http://neblora.fenst4r.life:4000/draw",
     "v2draw": "http://neblora.fenst4r.life:4000/v2draw",
-    "v3draw": "http://neblora.fenst4r.life:4000/v3draw",
-    "say": "http://neblora.fenst4r.life:4000/say",
-    "file": "https://fenst4r.life/api/file",
+    "analyze_image": "http://neblora.fenst4r.life:4848/analyze-image",
+    "tts": "http://neblora.fenst4r.life:4949/tts",
     "stt": "http://neblora.fenst4r.life:5000/stt",
-    "tts": "http://neblora.fenst4r.life:4949/tts"
+    "file": "https://fenst4r.life/api/file"
 }
 
-# ===== Функции API =====
-def chat_api(text, model="GPT-4", tts=False):
-    payload = {"text": text, "model": model}
-    r = requests.post(URLS["chatgpt"], json=payload)
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}", None
-    answer = r.json().get("response", "")
-    
-    audio_file = None
-    if tts:
-        tts_payload = {"text": answer, "voice":"nova"}
-        tts_r = requests.post(URLS["tts"], json=tts_payload)
-        if tts_r.status_code == 200:
-            audio_file = "output.mp3"
-            with open(audio_file, "wb") as f:
-                f.write(tts_r.content)
-    return answer, audio_file
-
-def llama_api(text):
-    r = requests.post(URLS["llama"], json={"text": text})
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}"
-    return r.json().get("response", "")
+HEADERS_JSON = {"Content-Type": "application/json"}
 
-def analyze_image_api(image):
-    r = requests.post(URLS["analyze_image"], files={"file": image})
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}"
-    return r.json().get("description", "")
+# ====== NEIROST4R CALL ======
+def call_neirost4r(api_version, message, profile="friendly", flags=None):
+    if flags is None:
+        flags = {}
+    url = NEIROST4R_BASE.get(api_version)
+    if not url:
+        return "Unknown NEIROST4R API version"
+    headers = HEADERS_JSON.copy()
+    # v4 и v5 требуют User-Agent
+    if api_version in ["ai_v4", "ai_v5"]:
+        headers["User-Agent"] = "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"
+    data = {"message": message, "profile": profile}
+    if api_version in ["ai_v2", "ai_v3", "ai_v4", "ai_v5"]:
+        data["flags"] = flags
+    try:
+        response = requests.post(url, headers=headers, data=json.dumps(data), timeout=30)
+        return response.json()
+    except Exception as e:
+        return {"error": str(e)}
 
-def draw_api(prompt, size="512x512"):
-    r = requests.post(URLS["draw"], json={"prompt": prompt, "size": size})
-    if r.status_code != 200:
-        return None
-    with open("draw_result.png", "wb") as f:
-        f.write(r.content)
-    return "draw_result.png"
+# ====== NEBLORA CALL ======
+def call_neblora(endpoint, text, model=None):
+    url = NEBLORA_BASE.get(endpoint)
+    if not url:
+        return "Unknown Neblora endpoint"
+    data = {"text": text}
+    if model:
+        data["model"] = model
+    try:
+        response = requests.post(url, headers=HEADERS_JSON, data=json.dumps(data), timeout=30)
+        return response.json()
+    except Exception as e:
+        return {"error": str(e)}
 
-def v2draw_api(prompt, format="json"):
-    r = requests.post(URLS["v2draw"], json={"prompt": prompt, "format": format})
-    if r.status_code != 200:
-        return "Ошибка"
-    if format=="json":
-        data = r.json()
-        return [img for img in data.get("images", [])]
-    return r.content
+# ====== Gradio Interface ======
+def generate_response(api_type, endpoint, text, profile, model, flags_json):
+    flags = {}
+    if flags_json.strip():
+        try:
+            flags = json.loads(flags_json)
+        except:
+            return "Flags must be valid JSON"
+    if api_type == "NEIROST4R":
+        return call_neirost4r(endpoint, text, profile=profile, flags=flags)
+    else:
+        return call_neblora(endpoint, text, model=model)
 
-def v3draw_api(prompt, width=1280, height=1280, steps=20, scale=7.5, format="file"):
-    payload = {"prompt": prompt, "width": width, "height": height, "steps": steps, "scale": scale, "format": format}
-    r = requests.post(URLS["v3draw"], json=payload)
-    if r.status_code != 200:
-        return None
-    if format=="json":
-        return r.json().get("image")
-    with open("v3draw.png", "wb") as f:
-        f.write(r.content)
-    return "v3draw.png"
-
-def say_api(text):
-    r = requests.post(URLS["say"], json={"text": text})
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}"
-    return r.json().get("text", "")
-
-def stt_api(audio_file):
-    r = requests.post(URLS["stt"], files={"file": audio_file})
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}"
-    return r.json().get("text", "")
-
-def upload_file(files):
-    payload = {}
-    for file in files:
-        payload[file.name] = file.read().decode('utf-8')
-    r = requests.post(URLS["file"], json=payload)
-    if r.status_code != 200:
-        return f"Ошибка: {r.status_code}"
-    return r.json().get("url", "")
-
-# ===== Интерфейс Gradio =====
 with gr.Blocks() as demo:
-    gr.Markdown("# NEIROST4R API Interface")
-    with gr.Tab("ChatGPT / TTS"):
-        text_input = gr.Textbox(label="Введите текст")
-        model_choice = gr.Radio(choices=["GPT-4", "GPT-5"], value="GPT-4", label="Модель")
-        tts_toggle = gr.Checkbox(label="Озвучить ответ")
-        chat_output = gr.Textbox(label="Ответ ChatGPT")
-        audio_output = gr.Audio(label="TTS", type="filepath")
-        gr.Button("Отправить").click(chat_api, inputs=[text_input, model_choice, tts_toggle], outputs=[chat_output, audio_output])
+    gr.Markdown("## NEIROST4R + Neblora Unified Interface")
     
-    with gr.Tab("LLaMA 3.1"):
-        llama_input = gr.Textbox(label="Введите текст")
-        llama_output = gr.Textbox(label="Ответ LLaMA")
-        gr.Button("Отправить").click(llama_api, inputs=[llama_input], outputs=[llama_output])
+    with gr.Row():
+        api_type = gr.Radio(["NEIROST4R", "Neblora"], label="Выберите API")
+        endpoint = gr.Dropdown(choices=[
+            "ai_v1","ai_v2","ai_v3","ai_v4","ai_v5",
+            "chatgpt","llama","draw","v2draw","analyze_image","tts","stt","file"
+        ], label="Endpoint / Модель")
     
-    with gr.Tab("Анализ изображения"):
-        image_input = gr.File(label="Выберите изображение", file_types=[".jpg", ".png", ".webp"])
-        image_desc = gr.Textbox(label="Описание изображения")
-        gr.Button("Анализировать").click(analyze_image_api, inputs=[image_input], outputs=[image_desc])
+    text_input = gr.Textbox(label="Введите текст", lines=4)
+    profile_input = gr.Textbox(label="Profile (для NEIROST4R)", value="friendly")
+    model_input = gr.Textbox(label="Model (для Neblora, если нужно)", value="")
+    flags_input = gr.Textbox(label="Flags JSON (для NEIROST4R)", value="{}", lines=4)
     
-    with gr.Tab("Draw / v2draw / v3draw"):
-        prompt_input = gr.Textbox(label="Описание сцены")
-        draw_out = gr.Image(label="Результат Draw")
-        gr.Button("Draw").click(draw_api, inputs=[prompt_input], outputs=[draw_out])
+    output = gr.JSON(label="Ответ")
     
-    with gr.Tab("Say / STT"):
-        say_input = gr.Textbox(label="Введите текст")
-        say_output = gr.Textbox(label="Текстовое преобразование")
-        gr.Button("Say").click(say_api, inputs=[say_input], outputs=[say_output])
-        
-        stt_input = gr.File(label="Аудиофайл")
-        stt_output = gr.Textbox(label="Распознанный текст")
-        gr.Button("STT").click(stt_api, inputs=[stt_input], outputs=[stt_output])
+    submit_btn = gr.Button("Отправить")
+    submit_btn.click(
+        generate_response,
+        inputs=[api_type, endpoint, text_input, profile_input, model_input, flags_input],
+        outputs=output
+    )
     
-    with gr.Tab("Файлы"):
-        file_input = gr.File(label="Загрузите файлы", file_types=[".txt", ".json"], file_types_allow_multiple=True)
-        file_url = gr.Textbox(label="Ссылка на файлы")
-        gr.Button("Загрузить").click(upload_file, inputs=[file_input], outputs=[file_url])
+    # ====== Файлы ======
+    file_input = gr.File(label="Загрузите файлы", file_types=[".txt", ".json"])
+    file_url = gr.Textbox(label="Ссылка на загруженный файл")
     
-    with gr.Tab("Документация"):
-        gr.Markdown("""
-        ## Содержание
-        - Инструкция по API NEIROST4R
-        - Быстрый старт
-        - API версии 1-5
-        - Python примеры
-        - Библиотека neirost4r
-        - Профили
-        - Флаги
-        - Примеры
-        - Провайдеры и модели
-        - Обработка ошибок
-
-        Новости в канале PUR🩷 CHANNEL: https://t.me/pur_channel
-        Поддержка и баги: @error_kill
-        """)
+    def upload_file(file):
+        if not file:
+            return "Нет файла"
+        try:
+            with open(file.name, "rb") as f:
+                files = {"file": f}
+                resp = requests.post(NEBLORA_BASE["file"], files=files)
+            return resp.json()
+        except Exception as e:
+            return {"error": str(e)}
+    
+    file_btn = gr.Button("Загрузить")
+    file_btn.click(upload_file, inputs=file_input, outputs=file_url)
 
 demo.launch()