Update app.py

app.py

@@ -1,166 +1,194 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
 import warnings
-import numpy as np
-from PIL import Image
 warnings.filterwarnings('ignore')
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-# Dicionário expandido de modelos
-models = {
-    'transcription': pipeline("automatic-speech-recognition", model="openai/whisper-small", device=device),
-    'translation': pipeline("translation", model="facebook/mbart-large-50-many-to-many-mmt", device=device),
-    'summarization': pipeline("summarization", model="facebook/bart-large-cnn", device=device),
-    'sentiment': pipeline("sentiment-analysis", model="nlptown/bert-base-multilingual-uncased-sentiment", device=device),
-    'question_answering': pipeline("question-answering", model="deepset/roberta-base-squad2", device=device),
-    'chat': pipeline("text-generation", model="facebook/opt-125m", device=device),
-    'image_caption': pipeline("image-to-text", model="Salesforce/blip-image-captioning-base", device=device),
-    'text_to_speech': pipeline("text-to-audio", model="facebook/mms-tts-eng", device=device),
-    'zero_shot': pipeline("zero-shot-classification", model="facebook/bart-large-mnli", device=device),
-    'ner': pipeline("token-classification", model="dslim/bert-base-NER", device=device)
+# Configurações
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+MAX_LENGTH = 2048
+
+# Usar modelo público e leve
+model_name = "microsoft/DialoGPT-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Tópicos polêmicos com versículos
+CONTROVERSIAL_TOPICS = {
+    "Sexualidade e Identidade": {
+        "description": "Perspectivas bíblicas sobre sexualidade, identidade e relacionamentos",
+        "verses": [
+            "Gênesis 1:27 - Criou Deus o homem à sua imagem...",
+            "1 Coríntios 6:9-11 - Não sabeis que os injustos...",
+            "Gálatas 3:28 - Não há judeu nem grego..."
+        ]
+    },
+    
+    "Aborto e Vida": {
+        "description": "Visão bíblica sobre a santidade da vida",
+        "verses": [
+            "Salmos 139:13-16 - Tu formaste o meu interior...",
+            "Jeremias 1:5 - Antes que te formasses...",
+            "Êxodo 20:13 - Não matarás"
+        ]
+    },
+    
+    "Riqueza e Pobreza": {
+        "description": "Perspectivas sobre desigualdade social",
+        "verses": [
+            "1 Timóteo 6:10 - O amor ao dinheiro é raiz...",
+            "Mateus 19:24 - É mais fácil passar um camelo...",
+            "Provérbios 22:2 - O rico e o pobre se encontram..."
+        ]
+    },
+    
+    "Política e Governo": {
+        "description": "Relação entre fé e política",
+        "verses": [
+            "Romanos 13:1-7 - Todo homem esteja sujeito...",
+            "Mateus 22:21 - Dai a César o que é de César...",
+            "1 Pedro 2:13-17 - Sujeitai-vos a toda autoridade..."
+        ]
+    },
+    
+    "Meio Ambiente": {
+        "description": "Mordomia ambiental",
+        "verses": [
+            "Gênesis 1:28 - Dominai sobre ela...",
+            "Gênesis 2:15 - Para o cultivar e o guardar...",
+            "Salmos 24:1 - Do Senhor é a terra..."
+        ]
+    },
+    
+    "Guerra e Paz": {
+        "description": "Perspectivas sobre conflito",
+        "verses": [
+            "Mateus 5:9 - Bem-aventurados os pacificadores...",
+            "Romanos 12:18 - Quanto depender de vós, tende paz...",
+            "Eclesiastes 3:8 - Tempo de guerra, e tempo de paz..."
+        ]
+    },
+    
+    "Ciência e Fé": {
+        "description": "Relação entre conhecimento e fé",
+        "verses": [
+            "Colossenses 2:8 - Cuidado que ninguém vos engane...",
+            "Romanos 1:20 - Seus atributos invisíveis...",
+            "Salmos 19:1 - Os céus proclamam..."
+        ]
+    },
+    
+    "Divórcio": {
+        "description": "Visões sobre matrimônio",
+        "verses": [
+            "Mateus 19:8-9 - Por causa da dureza...",
+            "Malaquias 2:16 - Porque o Senhor odeia o divórcio...",
+            "1 Coríntios 7:15 - Mas, se o descrente..."
+        ]
+    },
+    
+    "Papel da Mulher": {
+        "description": "Questões de gênero e liderança",
+        "verses": [
+            "Gálatas 3:28 - Não há homem nem mulher...",
+            "Provérbios 31 - Mulher virtuosa...",
+            "Juízes 4:4 - Débora, profetisa..."
+        ]
+    },
+    
+    "Livre Arbítrio": {
+        "description": "Escolha e soberania",
+        "verses": [
+            "João 3:16 - Para que todo aquele que nele crê...",
+            "Romanos 8:29-30 - Porquanto aos que conheceu...",
+            "2 Pedro 3:9 - Não querendo que ninguém pereça..."
+        ]
+    }
 }
 
-def process_transcription(audio):
-    if not audio:
-        return "Por favor, forneça um arquivo de áudio."
-    return models['transcription'](audio)["text"]
-
-def process_translation(text, direction):
-    if not text:
-        return "Por favor, forneça um texto para tradução."
-    return models['translation'](text, 
-        src_lang="pt" if direction=="pt_en" else "en", 
-        tgt_lang="en" if direction=="pt_en" else "pt")[0]['translation_text']
-
-def process_summarization(text):
-    if not text:
-        return "Por favor, forneça um texto para resumir."
-    return models['summarization'](text, max_length=130, min_length=30)[0]['summary_text']
-
-def process_sentiment(text):
-    if not text:
-        return "Por favor, forneça um texto para análise."
-    result = models['sentiment'](text)[0]
-    return f"Sentimento: {result['label']} (Score: {result['score']:.2f})"
-
-def process_qa(question, context):
-    if not question or not context:
-        return "Por favor, forneça tanto a pergunta quanto o contexto."
-    return models['question_answering'](question=question, context=context)['answer']
-
-def process_chat(message, history):
-    if not message:
-        return "", history
-    response = models['chat'](message, max_length=100, do_sample=True)[0]['generated_text']
-    history = history + [(message, response)]
-    return "", history
-
-def process_image_caption(image):
-    if image is None:
-        return "Por favor, forneça uma imagem."
-    return models['image_caption'](image)[0]['generated_text']
-
-def process_tts(text):
-    if not text:
-        return None
-    return models['text_to_speech'](text)
-
-def process_zero_shot(text, labels):
-    if not text or not labels:
-        return "Por favor, forneça texto e categorias."
-    result = models['zero_shot'](text, labels.split(','))
-    return f"Classificação: {result['labels'][0]} (Confiança: {result['scores'][0]:.2f})"
-
-def process_ner(text):
-    if not text:
-        return "Por favor, forneça um texto para análise."
-    results = models['ner'](text)
-    entities = []
-    for result in results:
-        entities.append(f"{result['word']}: {result['entity']}")
-    return "\n".join(entities)
-
+def process_message(message, history, topic=None):
+    try:
+        if topic:
+            info = CONTROVERSIAL_TOPICS[topic]
+            response = f"""
+            📚 Tema: {topic}
+            
+            📖 Descrição:
+            {info['description']}
+            
+            ✝️ Versículos Relevantes:
+            """
+            for verse in info['verses']:
+                response += f"• {verse}\n"
+            
+            response += "\nEstes versículos nos mostram diferentes aspectos deste tema. É importante estudá-los em seu contexto completo para uma melhor compreensão."
+            
+            return response
+        else:
+            # Para perguntas gerais
+            response = f"Sua pergunta: {message}\n\n"
+            response += "Pesquisando nas escrituras...\n"
+            # Procurar por palavras-chave e retornar versículos relevantes
+            for topic, info in CONTROVERSIAL_TOPICS.items():
+                if any(word.lower() in message.lower() for word in topic.split()):
+                    response += f"\nVersículos relacionados a {topic}:\n"
+                    for verse in info['verses'][:2]:
+                        response += f"• {verse}\n"
+            return response
+            
+    except Exception as e:
+        return f"Ocorreu um erro: {str(e)}"
+
+# Interface Gradio
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML(open("index.html").read())
+    gr.HTML("""
+        <div style="text-align: center; padding: 20px;">
+            <h1>📖 Análise Bíblica de Temas Contemporâneos</h1>
+            <p>Explore diferentes perspectivas bíblicas sobre questões atuais</p>
+        </div>
+    """)
     
-    with gr.Tabs():
-        # Aba de Processamento de Áudio
-        with gr.TabItem("🎤 Áudio"):
-            with gr.Tabs():
-                with gr.TabItem("Transcrição"):
-                    audio_input = gr.Audio(type="filepath")
-                    transcribe_button = gr.Button("Transcrever")
-                    transcription_output = gr.Textbox(label="Resultado")
-                    transcribe_button.click(process_transcription, inputs=audio_input, outputs=transcription_output)
-                
-                with gr.TabItem("Texto para Fala"):
-                    tts_input = gr.Textbox(label="Texto para Converter")
-                    tts_button = gr.Button("Converter para Áudio")
-                    tts_output = gr.Audio(label="Áudio Gerado")
-                    tts_button.click(process_tts, inputs=tts_input, outputs=tts_output)
+    with gr.Row():
+        # Coluna principal do chat
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(
+                height=600,
+                bubble_full_width=False,
+                show_label=False
+            )
+            with gr.Row():
+                msg = gr.Textbox(
+                    show_label=False,
+                    placeholder="Digite sua pergunta ou clique em um dos temas...",
+                    scale=4
+                )
+                clear = gr.Button("🗑️ Limpar", scale=1)
         
-        # Aba de Processamento de Texto
-        with gr.TabItem("📝 Texto"):
-            with gr.Tabs():
-                with gr.TabItem("Tradução"):
-                    with gr.Row():
-                        text_to_translate = gr.Textbox(label="Texto")
-                        translation_direction = gr.Radio(["en_pt", "pt_en"], value="en_pt", label="Direção")
-                    translate_button = gr.Button("Traduzir")
-                    translation_output = gr.Textbox(label="Resultado")
-                    translate_button.click(process_translation, inputs=[text_to_translate, translation_direction], outputs=translation_output)
-                
-                with gr.TabItem("Resumo"):
-                    text_sum = gr.Textbox(label="Texto", lines=5)
-                    sum_button = gr.Button("Resumir")
-                    sum_output = gr.Textbox(label="Resultado")
-                    sum_button.click(process_summarization, inputs=text_sum, outputs=sum_output)
-                
-                with gr.TabItem("Perguntas e Respostas"):
-                    qa_context = gr.Textbox(label="Contexto", lines=5)
-                    qa_question = gr.Textbox(label="Pergunta")
-                    qa_button = gr.Button("Obter Resposta")
-                    qa_output = gr.Textbox(label="Resposta")
-                    qa_button.click(process_qa, inputs=[qa_question, qa_context], outputs=qa_output)
-
-        # Aba de Análise
-        with gr.TabItem("📊 Análise"):
-            with gr.Tabs():
-                with gr.TabItem("Sentimento"):
-                    text_sent = gr.Textbox(label="Texto")
-                    sent_button = gr.Button("Analisar Sentimento")
-                    sent_output = gr.Textbox(label="Resultado")
-                    sent_button.click(process_sentiment, inputs=text_sent, outputs=sent_output)
-                
-                with gr.TabItem("Entidades Nomeadas"):
-                    ner_input = gr.Textbox(label="Texto")
-                    ner_button = gr.Button("Identificar Entidades")
-                    ner_output = gr.Textbox(label="Entidades Identificadas")
-                    ner_button.click(process_ner, inputs=ner_input, outputs=ner_output)
-                
-                with gr.TabItem("Classificação Zero-Shot"):
-                    zero_text = gr.Textbox(label="Texto")
-                    zero_labels = gr.Textbox(label="Categorias (separadas por vírgula)")
-                    zero_button = gr.Button("Classificar")
-                    zero_output = gr.Textbox(label="Resultado")
-                    zero_button.click(process_zero_shot, inputs=[zero_text, zero_labels], outputs=zero_output)
-
-        # Aba de IA Avançada
-        with gr.TabItem("🤖 IA Avançada"):
-            with gr.Tabs():
-                with gr.TabItem("Chat"):
-                    chatbot = gr.Chatbot()
-                    msg = gr.Textbox(label="Mensagem")
-                    clear = gr.Button("Limpar Conversa")
-                    msg.submit(process_chat, inputs=[msg, chatbot], outputs=[msg, chatbot])
-                    clear.click(lambda: None, None, chatbot, queue=False)
-                
-                with gr.TabItem("Descrição de Imagens"):
-                    image_input = gr.Image()
-                    caption_button = gr.Button("Gerar Descrição")
-                    caption_output = gr.Textbox(label="Descrição")
-                    caption_button.click(process_image_caption, inputs=image_input, outputs=caption_output)
-
-demo.launch(share=True)
+        # Coluna de tópicos
+        with gr.Column(scale=1):
+            gr.Markdown("### 🔍 Temas para Explorar")
+            
+            # Criar botões em grupos temáticos
+            for topic in CONTROVERSIAL_TOPICS.keys():
+                with gr.Box():
+                    btn = gr.Button(f"📚 {topic}", scale=1)
+                    btn.click(
+                        process_message,
+                        inputs=[gr.Textbox(value="", visible=False), chatbot],
+                        outputs=chatbot,
+                        kwargs={"topic": topic}
+                    )
+    
+    # Eventos
+    msg.submit(process_message, [msg, chatbot], [chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+    gr.Markdown("""
+    ### ℹ️ Como usar:
+    1. Clique em um dos temas para ver versículos relacionados
+    2. Digite uma pergunta específica no campo de texto
+    3. Use o botão 'Limpar' para reiniciar a conversa
+    """)
+
+# Iniciar a interface
+demo.launch(share=False)