Spaces:

DHEIVER
/

IA-Toolbox-Hub

Sleeping

App Files Files Community

DHEIVER commited on Nov 23, 2024

Commit

73cee64

verified ·

1 Parent(s): 9b185ce

Update app.py

Browse files

Files changed (1) hide show

app.py +112 -92

app.py CHANGED Viewed

@@ -1,98 +1,118 @@
-import gradio as gr
-from transformers import pipeline
-import torch
-import warnings
-warnings.filterwarnings('ignore')
-device = "cuda" if torch.cuda.is_available() else "cpu"
-# Expanded models dictionary with more services
-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)
-}
-def process_request(service_type, *args, **kwargs):
-    try:
-        if not args[0]: return "Por favor, forneça os dados necessários."
-        processors = {
-            'transcription': lambda audio: models['transcription'](audio)["text"],
-            'translation': lambda text, direction: models['translation'](text,
-                src_lang="pt" if direction=="pt_en" else "en",
-                tgt_lang="en" if direction=="pt_en" else "pt")[0]['translation_text'],
-            'summarization': lambda text: models['summarization'](text, max_length=130, min_length=30)[0]['summary_text'],
-            'sentiment': lambda text: models['sentiment'](text)[0]['label'],
-            'qa': lambda q, ctx: models['question_answering'](question=q, context=ctx)['answer'],
-            'chat': lambda msg, history: (models['chat'](msg, max_length=100, do_sample=True)[0]['generated_text'], history + [(msg, "")]),
-            'image_caption': lambda img: models['image_caption'](img)[0]['generated_text'],
-            'tts': lambda text: models['text_to_speech'](text),
-            'zero_shot': lambda text, labels: models['zero_shot'](text, labels)['labels'][0],
-            'ner': lambda text: models['ner'](text)
         }
-        return processors[service_type](*args)
-    except Exception as e:
-        return f"Erro: {str(e)}"
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.HTML(open("index.html").read())
-    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(lambda x: process_request('transcription', x), inputs=audio_input, outputs=transcription_output)
-        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")
-            translate_button = gr.Button("Traduzir")
-            translation_output = gr.Textbox(label="Resultado")
-            translate_button.click(lambda x, y: process_request('translation', x, y),
-                                inputs=[text_to_translate, translation_direction],
-                                outputs=translation_output)
-        with gr.TabItem("📊 Análise"):
-            with gr.Tabs():
-                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(lambda x: process_request('summarization', x), inputs=text_sum, outputs=sum_output)
-                with gr.TabItem("Sentimento"):
-                    text_sent = gr.Textbox(label="Texto")
-                    sent_button = gr.Button("Analisar")
-                    sent_output = gr.Textbox(label="Resultado")
-                    sent_button.click(lambda x: process_request('sentiment', x), inputs=text_sent, outputs=sent_output)
-        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")
-                    msg.submit(lambda x, y: process_request('chat', x, y), inputs=[msg, chatbot], outputs=[msg, chatbot])
-                    clear.click(lambda: None, None, chatbot, queue=False)
-                with gr.TabItem("Classificação"):
-                    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(lambda x, y: process_request('zero_shot', x, y.split(',')),
-                                    inputs=[zero_text, zero_labels],
-                                    outputs=zero_output)
-demo.launch(share=True)

+<!DOCTYPE html>
+<html lang="pt-BR">
+<head>
+    <meta charset="UTF-8">
+    <title>Central de IA - Bem-vindo</title>
+    <style>
+        .container {
+            max-width: 800px;
+            margin: 2rem auto;
+            padding: 0 1rem;
+            font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, sans-serif;
+        }
+        .header {
+            text-align: center;
+            margin-bottom: 3rem;
+            padding: 2rem;
+            background: linear-gradient(135deg, #6366f1 0%, #4f46e5 100%);
+            border-radius: 1rem;
+            color: white;
+        }
+        .header h1 {
+            margin: 0;
+            font-size: 2.5rem;
+            font-weight: 700;
+        }
+        .header p {
+            margin: 1rem 0 0;
+            font-size: 1.1rem;
+            opacity: 0.9;
+        }
+        .features {
+            display: grid;
+            grid-template-columns: repeat(auto-fit, minmax(250px, 1fr));
+            gap: 2rem;
+            margin: 3rem 0;
+        }
+        .feature-card {
+            padding: 1.5rem;
+            background: white;
+            border-radius: 0.75rem;
+            box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+            transition: transform 0.2s;
+        }
+        .feature-card:hover {
+            transform: translateY(-4px);
+        }
+        .feature-icon {
+            font-size: 2rem;
+            margin-bottom: 1rem;
+        }
+        .feature-title {
+            font-size: 1.25rem;
+            font-weight: 600;
+            margin-bottom: 0.5rem;
+            color: #1f2937;
+        }
+        .feature-description {
+            color: #6b7280;
+            line-height: 1.5;
+        }
+        .footer {
+            text-align: center;
+            margin-top: 3rem;
+            padding: 2rem;
+            color: #6b7280;
+            border-top: 1px solid #e5e7eb;
         }
+    </style>
+</head>
+<body>
+    <div class="container">
+        <header class="header">
+            <h1>Central de Inteligência Artificial</h1>
+            <p>Uma plataforma completa com ferramentas de IA para suas necessidades</p>
+        </header>
+        <div class="features">
+            <div class="feature-card">
+                <div class="feature-icon">📝</div>
+                <h3 class="feature-title">Transcrição de Áudio</h3>
+                <p class="feature-description">Converta áudio em texto com precisão usando tecnologia de ponta em reconhecimento de fala.</p>
+            </div>
+            <div class="feature-card">
+                <div class="feature-icon">🔄</div>
+                <h3 class="feature-title">Tradução</h3>
+                <p class="feature-description">Traduza textos entre português e inglês com nossa ferramenta de tradução neural.</p>
+            </div>
+            <div class="feature-card">
+                <div class="feature-icon">📊</div>
+                <h3 class="feature-title">Análise de Texto</h3>
+                <p class="feature-description">Gere resumos automáticos e analise o sentimento de textos com facilidade.</p>
+            </div>
+            <div class="feature-card">
+                <div class="feature-icon">🤖</div>
+                <h3 class="feature-title">IA Avançada</h3>
+                <p class="feature-description">Experimente nosso chatbot inteligente e classificação automática de textos.</p>
+            </div>
+        </div>
+        <footer class="footer">
+            <p>Desenvolvido com tecnologia de ponta em Inteligência Artificial</p>
+        </footer>
+    </div>
+</body>
+</html>