Update app.py

app.py

@@ -0,0 +1,77 @@
+import gradio as gr
+import uuid
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
+from sqlalchemy.orm import sessionmaker, declarative_base
+from sqlalchemy import Column, String, DateTime, func, select
+import asyncio
+
+# --------- Banco de dados (Supabase) -------------
+import os
+from dotenv import load_dotenv
+load_dotenv()
+
+DATABASE_URL = os.getenv("DATABASE_URL")  # Supabase URL
+engine = create_async_engine(DATABASE_URL, echo=False)
+SessionLocal = sessionmaker(bind=engine, class_=AsyncSession, expire_on_commit=False)
+Base = declarative_base()
+
+class User(Base):
+    __tablename__ = "users"
+    id = Column(String, primary_key=True)
+    name = Column(String)
+    api_key = Column(String, unique=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+async def init_db():
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+asyncio.run(init_db())
+
+# --------- Modelo Gemma 270M ---------------------
+model_name = "google/gemma-3-270m-it"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+gen_pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+# --------- Funções da API ------------------------
+async def create_user(name: str):
+    key = str(uuid.uuid4())
+    async with SessionLocal() as session:
+        user = User(id=str(uuid.uuid4()), name=name, api_key=key)
+        session.add(user)
+        await session.commit()
+    return key
+
+async def check_key(key: str):
+    async with SessionLocal() as session:
+        result = await session.execute(select(User).where(User.api_key == key))
+        user = result.scalar_one_or_none()
+    return user is not None
+
+def generate_text(prompt, api_key):
+    # Verifica chave
+    loop = asyncio.get_event_loop()
+    valid = loop.run_until_complete(check_key(api_key))
+    if not valid:
+        return "API Key inválida!"
+    result = gen_pipe(prompt, max_length=100, do_sample=True)
+    return result[0]["generated_text"]
+
+# --------- Interface Gradio ----------------------
+with gr.Blocks() as demo:
+    gr.Markdown("## API Gemma 270M - Teste")
+    with gr.Tab("Gerar Texto"):
+        prompt_input = gr.Textbox(label="Prompt")
+        api_key_input = gr.Textbox(label="API Key")
+        output_text = gr.Textbox(label="Resultado")
+        generate_btn = gr.Button("Gerar")
+        generate_btn.click(fn=generate_text, inputs=[prompt_input, api_key_input], outputs=output_text)
+    
+    with gr.Tab("Criar API Key"):
+        name_input = gr.Textbox(label="Nome do Usuário")
+        key_output = gr.Textbox(label="Sua API Key")
+        create_btn = gr.Button("Criar")
+        create_btn.click(fn=lambda name: asyncio.get_event_loop().run_until_complete(create_user(name)), inputs=name_input, outputs=key_output)
+
+demo.launch()