Update app.py

app.py

@@ -1,77 +1,70 @@
-import gradio as gr
+import os
 import uuid
+import json
+import gradio as gr
+import requests
 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()
+# --------- Carregar secrets -------------
+load_dotenv()
+SUPABASE_URL = os.environ.get("SUPABASE_URL")  # Ex.: https://xyz.supabase.co
+SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
+HEADERS = {
+    "apikey": SUPABASE_KEY,
+    "Authorization": f"Bearer {SUPABASE_KEY}",
+    "Content-Type": "application/json"
+}
 
-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())
+# --------- Funções Supabase -------------
+def create_user(name: str):
+    api_key = str(uuid.uuid4())
+    data = {
+        "name": name,
+        "api_key": api_key
+    }
+    response = requests.post(f"{SUPABASE_URL}/rest/v1/users",
+                             headers=HEADERS,
+                             data=json.dumps(data))
+    if response.status_code in [200, 201]:
+        return api_key
+    else:
+        return f"Erro: {response.text}"
 
-async def init_db():
-    async with engine.begin() as conn:
-        await conn.run_sync(Base.metadata.create_all)
-asyncio.run(init_db())
+def check_key(api_key: str):
+    response = requests.get(f"{SUPABASE_URL}/rest/v1/users?api_key=eq.{api_key}",
+                            headers=HEADERS)
+    data = response.json()
+    return len(data) > 0
 
-# --------- Modelo Gemma 270M ---------------------
+# --------- Carregar 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
-
+# --------- Função de geração de texto -------------
 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:
+    if not check_key(api_key):
         return "API Key inválida!"
     result = gen_pipe(prompt, max_length=100, do_sample=True)
     return result[0]["generated_text"]
 
-# --------- Interface Gradio ----------------------
+# --------- Interface Gradio -------------
 with gr.Blocks() as demo:
-    gr.Markdown("## API Gemma 270M - Teste")
+    gr.Markdown("## Gemma 270M API - Espaço Gratuito")
+    
     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)
+        create_btn.click(fn=create_user, inputs=name_input, outputs=key_output)
 
-demo.launch()
+demo.launch(server_name="0.0.0.0", server_port=7860)