merged with github repo

README.md

@@ -9,7 +9,9 @@ app_file: run.py
 pinned: false
 ---
 
-# Программная инженерия. Практические задания №2, №3 и №4
+[![Tests](https://github.com/kavlab/urfu_iml_2023_1_3_hw2/actions/workflows/python-app.yml/badge.svg)](https://github.com/kavlab/urfu_iml_2023_1_3_hw2/actions/workflows/python-app.yml)
+
+# Программная инженерия. Практические задания №2, №3, №4 и №5
 
 Приложение разработано с использованием фреймворка [Streamlit](https://streamlit.io/).
 Состоит из двух страниц и Главной страницы:
@@ -48,7 +50,7 @@ streamlit run run.py
 ### Генератор аудио
 Необходимо ввести текст в текстовое поле и нажать кнопку "Генерировать!!!". В результате появится аудио запись на английском языке с описанием введенного текста.
 
-![Результат работы моделей "Генератор аудио / Текст"](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/audio_gen_image.png)
+![Результат работы моделей "Генератор аудио / Текст"](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/screens/audio_gen_image.png)
 ![Результат работы моделей "Генератор аудио / Аудио"](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/kuznetsov_av/text_to_speech_image.png)
 
 ### Описание изображения
@@ -65,7 +67,7 @@ streamlit run run.py
 #### Пример вызова сервиса Определение языка текста
 Вызвать url сервиса ```<host>/langdetector/detect``` методом POST
 
-![img.png](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/PostmanLangDetect.png)
+![img.png](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/screens/PostmanLangDetect.png)
 
 Передаваемые параметры:
 ```
@@ -89,7 +91,7 @@ streamlit run run.py
 #### Пример вызова сервиса Перевод текста с языка Ru на En
 Вызвать url сервиса ```<host>/translator/translate``` методом POST
 
-![img.png](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/PostmanTranslate.png)
+![img.png](https://raw.githubusercontent.com/kavlab/urfu_iml_2023_1_3_hw2/main/mulyavin_aa/screens/PostmanTranslate.png)
 
 Передаваемые параметры:
 ```
@@ -137,3 +139,5 @@ streamlit run run.py
     "url": "https://fikiwiki.com/uploads/posts/2022-02/1645000127_53-fikiwiki-com-p-kartinki-krasivie-babochki-narisovannie-55.png"
 }
 ```
+
+Тестирует GitHub Actions

kuznetsov_av/test_api.py

@@ -1,5 +1,5 @@
 from fastapi.testclient import TestClient
-from kuznetsov_av.api import Request, Response, app
+from kuznetsov_av.api import Request, app
 
 client = TestClient(app)
 

requirements.txt

@@ -10,4 +10,5 @@ fastapi==0.104.1
 uvicorn==0.24.0
 httpx==0.25.2
 pillow~=10.1.0
-pydantic~=2.5.2
+pydantic~=2.5.2
+requests~=2.31.0

api.py → run_api.py

@@ -1,3 +1,4 @@
+import os
 import uvicorn
 from fastapi import FastAPI
 
@@ -37,7 +38,7 @@ async def root():
                 },
                 {
                     "descr": "API для описания загруженного изображения",
-                    "base_url": "/get_description_image"
+                    "base_url": "/get_description_image/predict/"
                 }
             ]}
 
@@ -75,12 +76,11 @@ async def text_to_speech(entity: Request) -> Response:
 @app.post("/get_description_image/predict/")
 async def get_description_image_predict(item: develop_api_app.Url):
     """
-    Получение ссылки на изображение. Запись изображения в директорию zvereva_ev
-    под названием 'image.png для получения его описания
+    Получение ссылки на изображение. Запись изображения под названием 'image_predict_test_image.png для получения его описания
     """
     response = requests.get(item.url, stream=True)
     # сохранение изображения для дальнейшей передачи в модель
-    with open("image.png", "wb") as f:
+    with open(f"{os.path.dirname(__file__)}/zvereva_ev/image_predict_tmp.png", "wb") as f:
         f.write(response.content)
 
     return get_description_image.get_description_image()

test/test_api.py

@@ -0,0 +1,151 @@
+import kuznetsov_av
+from fastapi.testclient import TestClient
+from run_api import app
+
+client = TestClient(app)
+
+
+def test_get_description_image_predict() -> None:
+    """
+    Тест API Описание загруженного изображения
+    """
+    from zvereva_ev.develop_api_app import Url
+
+    response = client.post(
+        url="/get_description_image/predict/",
+        json=Url(url="https://fikiwiki.com/uploads/posts/2022-02/1645000127_53-"
+                     "fikiwiki-com-p-kartinki-krasivie-babochki"
+                     "-narisovannie-55.png"
+                 ).model_dump())
+
+    assert response.status_code == 200
+    assert response.text == '"Фото бабочки с оранжевыми крыльями и белыми точками"'
+
+
+def test_langdetector_api_ru() -> None:
+    """
+    Тест API определения языка текста RU (mulyavin_aa)
+    """
+    import mulyavin_aa.model.langdetector
+
+    api_resp = client.post(
+        url="/langdetector/detect",
+        json=mulyavin_aa.model.langdetector.Request(
+            text='Доброго дня всем котам!').model_dump())
+
+    response = mulyavin_aa.model.langdetector.Response.model_validate_json(api_resp.text)
+
+    assert api_resp.status_code == 200
+    assert len(response.langs) > 0
+    assert response.langs[0].label == 'ru'
+
+
+def test_langdetector_api_en() -> None:
+    """
+    Тест API определения языка текста EN (mulyavin_aa)
+    """
+    import mulyavin_aa.model.langdetector
+
+    api_resp = client.post(
+        url="/langdetector/detect",
+        json=mulyavin_aa.model.langdetector.Request(
+            text='So I checked functions in the class model').model_dump())
+
+    response = mulyavin_aa.model.langdetector.Response.model_validate_json(api_resp.text)
+
+    assert api_resp.status_code == 200
+    assert len(response.langs) > 0
+    assert response.langs[0].label == 'en'
+
+
+def test_langdetector_api_err() -> None:
+    """
+    Тест API определения языка текста не Ru и En (mulyavin_aa)
+    Модель не ограничена только Ru и En
+    """
+    import mulyavin_aa.model.langdetector
+
+    api_resp = client.post(
+        url="/langdetector/detect",
+        json=mulyavin_aa.model.langdetector.Request(
+            text='').model_dump())
+
+    response = mulyavin_aa.model.langdetector.Response.model_validate_json(api_resp.text)
+
+    assert api_resp.status_code == 200
+    assert len(response.langs) > 0
+    assert response.langs[0].label != 'en'
+    assert response.langs[0].label != 'ru'
+
+
+def test_translator_ru_to_en() -> None:
+    """
+    Тест API Перевод текста с языка Ru на En (mulyavin_aa)
+    """
+    import mulyavin_aa.model.translator
+
+    api_resp = client.post(
+        url="/translator/translate",
+        json=mulyavin_aa.model.translator.Request(
+            text='Доброго деня всем котам!').model_dump())
+
+    response = mulyavin_aa.model.translator.Response.model_validate_json(api_resp.text)
+
+    assert api_resp.status_code == 200
+    assert len(response.text) > 0
+    assert response.text == 'Good day to all cats!'
+
+
+def test_translator_en_to_en() -> None:
+    """
+    Тест API Перевод текста с языка En на En (mulyavin_aa)
+    """
+    import mulyavin_aa.model.translator
+
+    api_resp = client.post(
+        url="/translator/translate",
+        json=mulyavin_aa.model.translator.Request(
+            text='Good day to all cats!').model_dump())
+
+    response = mulyavin_aa.model.translator.Response.model_validate_json(api_resp.text)
+
+    assert api_resp.status_code == 200
+    assert len(response.text) > 0
+    assert response.text == 'Good day to all cats!'
+
+
+def test_text_to_speech_ok():
+    """
+    Тест API преобразования текста в речь
+    """
+    response = client.post(
+        url='/text-to-speech/convert/',
+        json=kuznetsov_av.api.Request(text='Test').model_dump()
+    )
+
+    assert response.status_code == 200
+    assert response.json().get('audio') is not None
+    assert type(response.json().get('audio')) == str
+    assert len(response.json().get('audio')) > 0
+    assert response.json().get('sampling_rate') is not None
+    assert type(response.json().get('sampling_rate')) == int
+
+
+def test_text_to_speech_error422():
+    """
+    Тест API преобразования текста в речь
+    """
+    response = client.post(
+        url='/text-to-speech/convert/',
+        json=''
+    )
+
+    assert response.status_code == 422
+
+
+def test_root():
+    response = client.get('/')
+
+    assert response.status_code == 200
+    assert response.json().get('message') is not None
+    assert len(response.json().get('message')) > 0

zvereva_ev/get_description_image.py

@@ -20,7 +20,7 @@ def load_image():
         st.image(image_data)
         image_data_open = Image.open(io.BytesIO(image_data))
         # Сохранение изображения из буффера
-        image_data_open.save("image.png")
+        image_data_open.save(f"{os.path.dirname(__file__)}/image_predict_tmp.png")
         return image_data_open
     else:
         return None
@@ -39,7 +39,7 @@ def get_description_image():
     """
     processor, model = load_models()
 
-    raw_image = Image.open("image.png")
+    raw_image = Image.open(f"{os.path.dirname(__file__)}/image_predict_tmp.png")
 
     text = "a photography of"
     inputs = processor(raw_image, text, return_tensors="pt")
@@ -53,7 +53,7 @@ def get_description_image():
 
 
 # Получение полного пути к сохраненному изображению из буффера
-file_path = "urfu_iml_2023_1_3_hw2/zvereva_ev/image.png"
+file_path = f"{os.path.dirname(__file__)}/image_predict_tmp.png"
 absolute_path = os.path.abspath(file_path)
 
 
@@ -71,7 +71,7 @@ def run_():
                 st.warning("Загрузите изображение, чтобы получить описание", icon="🚀")
         else:
             exit()
-        if "image.png" in absolute_path:
-            os.remove("image.png")
+        if f"{os.path.dirname(__file__)}/image_predict_tmp.png" in absolute_path:
+            os.remove(f"{os.path.dirname(__file__)}/image_predict_tmp.png")
     except FileNotFoundError:
         return "Изображение не загружено"