Update app.py

app.py

@@ -16,7 +16,9 @@ from fastapi import FastAPI, HTTPException, Query
 from typing import List
 import uvicorn
 from starlette.requests import Request
-from starlette.responses import HTMLResponse
+from starlette.responses import HTMLResponse, JSONResponse
+from fastapi.responses import HTMLResponse
+from fastapi.middleware.cors import CORSMiddleware
 
 # Настройка логирования
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -57,6 +59,15 @@ movies_table = "Movies"  # Имя таблицы с фильмами
 # FastAPI приложение
 app = FastAPI()
 
+# Разрешаем CORS, чтобы Gradio мог обращаться к API
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Разрешаем все источники, в продакшене лучше указать конкретные
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 def get_db_connection():
     """Устанавливает соединение с базой данных."""
     try:
@@ -159,7 +170,7 @@ def get_movie_data_from_db(conn, movie_ids):
         logging.error(f"Ошибка при получении данных фильмов из БД: {e}")
     return movie_data_dict
 
-def rerank_with_api(query, results, top_k):
+def rerank_with_api(query, results, top_k, rerank_top_k=None):
     """Переранжирует результаты с помощью Jina AI Reranker API."""
     logging.info(f"Начало переранжирования для запроса: '{query}'")
 
@@ -184,7 +195,7 @@ def rerank_with_api(query, results, top_k):
     data = {
         "model": JINA_RERANKER_MODEL,
         "query": query,
-        "top_n": top_k,
+        "top_n": rerank_top_k or top_k*2,
         "documents": documents
     }
     logging.info(f"Отправка данных на реранжировку (documents count): {len(data['documents'])}")
@@ -211,7 +222,7 @@ def rerank_with_api(query, results, top_k):
         logging.error(f"Ошибка при запросе к API реранжировщика: {e}")
         return []
 
-def search_movies_internal(query: str, top_k: int = 25):
+def search_movies_internal(query: str, top_k: int = 25, rerank_top_k: int = None):
     """Внутренняя функция для поиска фильмов по запросу (используется и в Gradio, и в API)."""
     start_time = time.time()
 
@@ -263,7 +274,7 @@ def search_movies_internal(query: str, top_k: int = 25):
             conn.close()
 
         # Переранжируем результаты с помощью API
-        reranked_results = rerank_with_api(query, results, top_k)
+        reranked_results = rerank_with_api(query, results, top_k, rerank_top_k)
 
         conn = get_db_connection()
         movie_ids = [movie_id for movie_id, _ in reranked_results]
@@ -295,41 +306,106 @@ def search_movies_internal(query: str, top_k: int = 25):
         logging.error(f"Ошибка при выполнении поиска: {e}")
         raise
 
-def search_movies(query, top_k=25):
-    """Функция поиска фильмов для Gradio интерфейса."""
-    try:
-        results, search_time = search_movies_internal(query, top_k)
-        output = f"<p>Время поиска: {search_time:.2f} сек</p>"
-        for result in results:
-            output += f"<h3>{result['name']} ({result['year']})</h3>\n"
-            output += f"<p><strong>Жанры:</strong> {', '.join(result['genres'])}</p>\n"
-            output += f"<p><strong>Описание:</strong> {result['description']}</p>\n"
-            output += f"<p><strong>Релевантность (reranker score):</strong> {result['relevance_score']:.4f}</p>\n"
-            output += "<hr>\n"
-        return output
-    except Exception as e:
-        return f"<p>Произошла ошибка при выполнении поиска: {e}</p>"
-
 @app.get("/search/", response_model=List[dict])
-async def api_search_movies(query: str = Query(..., description="Поисковый запрос"), top_k: int = Query(25, description="Количество возвращаемых результатов")):
+async def api_search_movies(query: str = Query(..., description="Поисковый запрос"), 
+                            top_k: int = Query(25, description="Количество возвращаемых результатов"),
+                            rerank_top_k: int = Query(None, description="Количество фильмов для передачи в реранкер (если не указано, то top_k*2)")):
     """API endpoint для поиска фильмов."""
     try:
-        results, _ = search_movies_internal(query, top_k)
+        results, _ = search_movies_internal(query, top_k, rerank_top_k)
         return results
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# Создаем интерфейс Gradio
-iface = gr.Interface(
-    fn=search_movies,
-    inputs=gr.Textbox(lines=2, placeholder="Введите запрос для поиска фильмов..."),
-    outputs=gr.HTML(label="Результаты поиска"),
-    title="Семантический поиск фильмов",
-    description="Введите описание фильма, который вы ищете, и система найдет наиболее похожие фильмы."
-)
-
-# Встраиваем Gradio в FastAPI
-app = gr.mount_gradio_app(app, iface, path="/")
+# Рут-эндпоинт, который отдаёт HTML-страницу
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    return """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Семантический поиск фильмов</title>
+        <style>
+            body {
+                font-family: sans-serif;
+                margin: 20px;
+            }
+            h1 {
+                text-align: center;
+            }
+            label {
+                display: block;
+                margin-bottom: 5px;
+            }
+            input[type="text"] {
+                width: 100%;
+                padding: 10px;
+                margin-bottom: 10px;
+                border: 1px solid #ccc;
+                box-sizing: border-box;
+            }
+            button {
+                padding: 10px 20px;
+                background-color: #4CAF50;
+                color: white;
+                border: none;
+                cursor: pointer;
+            }
+            #results {
+                margin-top: 20px;
+            }
+            .movie {
+                border: 1px solid #ccc;
+                padding: 10px;
+                margin-bottom: 10px;
+            }
+            .movie h3 {
+                margin-top: 0;
+            }
+        </style>
+    </head>
+    <body>
+        <h1>Семантический поиск фильмов</h1>
+        <label for="query">Введите запрос:</label>
+        <input type="text" id="query" name="query" placeholder="Введите описание фильма...">
+        <button onclick="searchMovies()">Искать</button>
+        <div id="results"></div>
+
+        <script>
+            function searchMovies() {
+                const query = document.getElementById("query").value;
+                const resultsDiv = document.getElementById("results");
+                resultsDiv.innerHTML = "<p>Поиск...</p>";
+
+                fetch(`/search/?query=${encodeURIComponent(query)}`)
+                    .then(response => response.json())
+                    .then(data => {
+                        resultsDiv.innerHTML = "";
+                        if (data.length === 0) {
+                            resultsDiv.innerHTML = "<p>Ничего не найдено.</p>";
+                        } else {
+                            data.forEach(movie => {
+                                const movieDiv = document.createElement("div");
+                                movieDiv.classList.add("movie");
+                                movieDiv.innerHTML = `
+                                    <h3>${movie.name} (${movie.year})</h3>
+                                    <p><strong>Жанры:</strong> ${movie.genres.join(", ")}</p>
+                                    <p><strong>Описание:</strong> ${movie.description}</p>
+                                    <p><strong>Релевантность:</strong> ${movie.relevance_score.toFixed(4)}</p>
+                                `;
+                                resultsDiv.appendChild(movieDiv);
+                            });
+                        }
+                    })
+                    .catch(error => {
+                        console.error("Ошибка:", error);
+                        resultsDiv.innerHTML = "<p>Произошла ошибка при поиске.</p>";
+                    });
+            }
+        </script>
+    </body>
+    </html>
+    """
 
 # Рут-эндпоинт для демонстрации, что FastAPI работает
 @app.get("/api")