Spaces:

vioott
/

BookMatchAI-Python-GeminiAPI

Sleeping

App Files Files Community

vioott commited on Jul 22

Commit

5b5c1df

1 Parent(s): e9ba6dc

refactor: reorganized project by separating routes and services

Browse files

Files changed (8) hide show

app.py +7 -215
constants/genres.py +8 -0
routes/__init__.py +0 -0
routes/chat.py +71 -0
routes/home.py +31 -0
routes/profile.py +42 -0
services/__init__.py +0 -0
services/genai_service.py +27 -0

app.py CHANGED Viewed

@@ -1,221 +1,13 @@
-import os
-import re
-import google.generativeai as genai
-from flask import Flask, redirect, render_template, request
-from user_data import get_user_history, user_histories
-from logs import save_log
 app = Flask(__name__)
-# Função para configurar a chave de API
-def genai_configuration():
-    try:
-        genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
-        # Testa se a chave principal está disponível
-        test_model = genai.GenerativeModel("gemini-1.5-flash")
-        test_model.start_chat().send_message("ping")
-        return
-    except Exception:
-        print("⚠️  Falha com a chave principal, tentando chave reserva 1.")
-        try:
-            genai.configure(api_key=os.environ['GOOGLE_API_KEY_2'])
-            # Testa se a chave reserva 1 está disponível
-            test_model = genai.GenerativeModel("gemini-1.5-flash")
-            test_model.start_chat().send_message("ping")
-            return
-        except Exception:
-            print("⚠️  Falha com a chave reserva 1, tentando chave reserva 2.")
-            try:
-                genai.configure(api_key=os.environ['GOOGLE_API_KEY_3'])
-                # Testa se a chave reserva 2 está disponível
-                test_model = genai.GenerativeModel("gemini-1.5-flash")
-                test_model.start_chat().send_message("ping")
-                return
-            except Exception:
-                print("❌ Todas as chaves falharam.")
-genai_configuration()
-# Definir o modelo generativo com as funções disponíveis
-magical_if = genai.GenerativeModel("gemini-1.5-flash")
-GENRES = sorted([
-    'ficção', 'não-ficção', 'ciência', 'fantasia', 'romance', 'mistério',
-    'terror', 'aventura', 'história', 'biografia', 'autoajuda',
-    'espiritualidade', 'filosofia', 'poesia', 'drama', 'humor', 'tecnologia',
-    'educação', 'negócios'
-])
-def ia_decision(user_id, history):
-    business_rules = """
-    Analise o histórico de compras do usuário e recomende livros a ele.
-    Regras:
-    1. Se o usuário demonstrou interesse em ficção, recomende ficção.
-    2. Se o usuário demonstrou interesse em não ficção, recomende não-ficção.
-    3. Se o usuário demonstrou interesse em ciência, recomende ciência.
-    4. Se o usuário demonstrou interesse em fantasia, recomende fantasia.
-    5. Se o usuário não demonstrou interesse em nenhuma categoria, recomende
-      os livros mais vendidos no último mês.
-    """
-    user_decision = magical_if.start_chat(
-        enable_automatic_function_calling=True
-    )
-    message = (
-        f"Histórico do usuário {user_id}: {history}\n"
-        f"Regras de negócio:\n{business_rules}"
-    )
-    response = user_decision.send_message(message)
-    return response.text
-@app.route('/', methods=['GET', 'POST'])
-def inicio():
-    if request.method == 'POST':
-        name = request.form.get('name')
-        preferences = request.form.getlist('preferences')
-        # Simula criação de novo ID
-        new_id = max(user_histories.keys()) + 1
-        user_histories[new_id] = {
-            'name': name,
-            'preferences': preferences
-        }
-        return redirect(f'/chat/{new_id}')
-    return render_template(
-        'start.html',
-        profiles=user_histories,
-        genres=GENRES
-    )
-@app.route('/perfil/<int:user_id>', methods=['GET', 'POST'])
-def perfil(user_id):
-    history = get_user_history(user_id)
-    if not history:
-        return '''
-            <html>
-            <head>
-                <meta http-equiv="refresh" content="2;url=/">
-            </head>
-            <body>
-                <h2>Ops, algo deu errado por aqui.</h2>
-                <p>Vou te levar de volta para a página inicial.</p>
-            </body>
-            </html>
-        '''
-    if request.method == 'POST':
-        name = request.form.get('name')
-        preferences = request.form.getlist('preferences')
-        # aqui vamos salvar nome e preferências
-        history['name'] = name
-        history['preferences'] = preferences
-        return redirect(f'/chat/{user_id}')
-    return render_template(
-        'profile.html',
-        user_id=user_id,
-        history=history,
-        genres=GENRES
-    )
-@app.route('/chat/<int:user_id>', methods=['GET', 'POST'])
-def chat(user_id):
-    history = get_user_history(user_id)
-    if not history:
-        return '''
-            <html>
-            <head>
-                <meta http-equiv="refresh" content="2;url=/">
-            </head>
-            <body>
-                <h2>Ops, algo deu errado por aqui.</h2>
-                <p>Vou te levar de volta para a página inicial.</p>
-            </body>
-            </html>
-        '''
-    response = None
-    if request.method == 'POST':
-        question = request.form.get('question')
-        user_decision = magical_if.start_chat(
-            enable_automatic_function_calling=True
-        )
-        name = history.get("name", f"Usuário {user_id}")
-        preferences = history.get("preferences", [])
-        prompt = (
-            f"Você é uma especialista em livros e pode conversar livremente "
-            f"sobre obras literárias, personagens, enredos, gêneros e autores."
-            f" O nome do usuário é {name} e seu ID é {user_id}. A informação "
-            f"sobre o id é somente para você, não mencione nada sobre isso ao "
-            f"usuário. As preferências dele são: "
-            f"{', '.join(preferences) or 'nenhuma preferência registrada'}. "
-            f"Histórico de compras: {history}. "
-            f"Pergunta: {question}"
-            f"Responda na mesma língua em que foi feita a pergunta."
-            f"Sempre fale com o usuário se dirigindo diretamente a ele, "
-            f"não se refira a ele em terceira pessoa, como se estivesse "
-            f"falando de outra pessoa."
-            f" Dê respostas com no máximo 500 caracteres."
-        )
-        gemini_response = user_decision.send_message(prompt)
-        response = gemini_response.text
-        save_log(user_id, history, response)
-    formatted_response = None
-    if response:
-        formatted_response = response.replace('\n', '<br>')
-        formatted_response = re.sub(
-            r'(?<=<br>)\s*(Ficção|Não[- ]Ficção|Ciência|Fantasia)\s*:(?=<br>)',
-            r'<b>\1:</b>',
-            f"<br>{formatted_response}<br>"
-        )[4:-4]
-        formatted_response = formatted_response.replace('**', '')
-        formatted_response = formatted_response.replace('* ', '• ')
-    return render_template(
-        'chat.html',
-        user_id=user_id,
-        response=formatted_response,
-        history=history
-    )
 if __name__ == '__main__':
     app.run(debug=True)

+from flask import Flask
+from routes.home import home_bp
+from routes.chat import chat_bp
+from routes.profile import profile_bp
 app = Flask(__name__)
+app.register_blueprint(home_bp)
+app.register_blueprint(chat_bp)
+app.register_blueprint(profile_bp)
 if __name__ == '__main__':
     app.run(debug=True)

constants/genres.py ADDED Viewed

	@@ -0,0 +1,8 @@

+# Lista de gêneros literários disponíveis para seleção de preferências.
+GENRES = sorted([
+    'ficção', 'não-ficção', 'ciência', 'fantasia', 'romance', 'mistério',
+    'terror', 'aventura', 'história', 'biografia', 'autoajuda',
+    'espiritualidade', 'filosofia', 'poesia', 'drama', 'humor',
+    'tecnologia', 'educação', 'negócios'
+])

routes/__init__.py ADDED Viewed

File without changes

routes/chat.py ADDED Viewed

	@@ -0,0 +1,71 @@

+import re
+from flask import Blueprint, render_template, request
+from services.genai_service import configure_genai
+from logs import get_user_history, save_log
+chat_bp = Blueprint('chat', __name__)
+model = configure_genai()
+@chat_bp.route('/chat/<int:user_id>', methods=['GET', 'POST'])
+def chat(user_id):
+    """
+    Rota para interação do usuário com a IA.
+    """
+    history = get_user_history(user_id)
+    if not history:
+        return '''
+            <html>
+            <head>
+                <meta http-equiv="refresh" content="2;url=/" />
+            </head>
+            <body>
+                <h2>Ops, algo deu errado por aqui.</h2>
+                <p>Vou te levar de volta para a página inicial.</p>
+            </body>
+            </html>
+        '''
+    response = None
+    if request.method == 'POST':
+        question = request.form.get('question')
+        session = model.start_chat(enable_automatic_function_calling=True)
+        prompt = (
+            f"Você é uma especialista em livros e pode conversar livremente "
+            f"sobre obras literárias, personagens, enredos, gêneros e autores. "
+            f"O nome do usuário é {history.get('name', f'Usuário {user_id}')}. "
+            f"As preferências dele são: "
+            f"{', '.join(history.get('preferences', [])) or 'nenhuma'}. "
+            f"Histórico: {history}. "
+            f"Pergunta: {question} "
+            f"Responda na mesma língua da pergunta, com até 500 caracteres. "
+            f"Fale diretamente com o usuário."
+        )
+        gemini_response = session.send_message(prompt)
+        response = gemini_response.text
+        save_log(user_id, history, response)
+    formatted_response = None
+    if response:
+        formatted_response = response.replace('\n', '<br>')
+        formatted_response = re.sub(
+            r'(?<=<br>)\s*(Ficção|Não[- ]Ficção|Ciência|Fantasia)\s*:(?=<br>)',
+            r'<b>\1:</b>',
+            f"<br>{formatted_response}<br>"
+        )[4:-4]
+        formatted_response = formatted_response.replace('**', '')
+        formatted_response = formatted_response.replace('* ', '• ')
+    return render_template(
+        'chat.html',
+        user_id=user_id,
+        response=formatted_response,
+        history=history
+    )

routes/home.py ADDED Viewed

	@@ -0,0 +1,31 @@

+from flask import Blueprint, render_template, request, redirect
+from logs import get_user_history
+from constants.genres import GENRES
+home_bp = Blueprint('home', __name__)
+@home_bp.route('/', methods=['GET', 'POST'])
+def inicio():
+    """
+    Rota inicial para criação de perfil ou acesso a perfis existentes.
+    """
+    if request.method == 'POST':
+        name = request.form.get('name')
+        preferences = request.form.getlist('preferences')
+        # Simula criação de novo ID com base no maior já existente
+        new_id = max(get_user_history.keys(), default=0) + 1
+        get_user_history[new_id] = {
+            'name': name,
+            'preferences': preferences
+        }
+        return redirect(f'/chat/{new_id}')
+    return render_template(
+        'start.html',
+        profiles=get_user_history,
+        genres=GENRES
+    )

routes/profile.py ADDED Viewed

	@@ -0,0 +1,42 @@

+from flask import Blueprint, render_template, request, redirect
+from logs import get_user_history
+from constants.genres import GENRES
+profile_bp = Blueprint('profile', __name__)
+@profile_bp.route('/perfil/<int:user_id>', methods=['GET', 'POST'])
+def perfil(user_id):
+    """
+    Rota para visualizar e atualizar nome e preferências do perfil.
+    """
+    history = get_user_history(user_id)
+    if not history:
+        return '''
+            <html>
+            <head>
+                <meta http-equiv="refresh" content="2;url=/" />
+            </head>
+            <body>
+                <h2>Ops, algo deu errado por aqui.</h2>
+                <p>Vou te levar de volta para a página inicial.</p>
+            </body>
+            </html>
+        '''
+    if request.method == 'POST':
+        name = request.form.get('name')
+        preferences = request.form.getlist('preferences')
+        history['name'] = name
+        history['preferences'] = preferences
+        return redirect(f'/chat/{user_id}')
+    return render_template(
+        'profile.html',
+        user_id=user_id,
+        history=history,
+        genres=GENRES
+    )

services/__init__.py ADDED Viewed

File without changes

services/genai_service.py ADDED Viewed

	@@ -0,0 +1,27 @@

+import os
+import google.generativeai as genai
+def configure_genai():
+    """
+    Configura a chave de API e inicializa o modelo Gemini.
+    """
+    api_key = os.environ.get("GOOGLE_API_KEY")
+    if not api_key:
+        raise EnvironmentError(
+            "❌ A variável de ambiente GOOGLE_API_KEY não está definida."
+        )
+    try:
+        genai.configure(api_key=api_key)
+        # Testa se a chave está funcionando
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        model.start_chat().send_message("ping")
+        print("✅ Chave da API configurada com sucesso.")
+        return model
+    except Exception as e:
+        raise RuntimeError(f"❌ Falha ao configurar a chave da API: {e}")