Update app.py

app.py

@@ -4,26 +4,58 @@ import torchaudio
 import numpy as np
 from transformers import AutoProcessor, SeamlessM4Tv2Model
 from datetime import datetime
+import time
+import threading
+import queue
+import sounddevice as sd
 
-class SeamlessTranslator:
+class ARISTranslator:
     def __init__(self, model_name: str = "facebook/seamless-m4t-v2-large"):
         self.processor = AutoProcessor.from_pretrained(model_name)
         self.model = SeamlessM4Tv2Model.from_pretrained(model_name)
         self.sample_rate = self.model.config.sampling_rate
+        self.audio_queue = queue.Queue()
+        self.is_recording = False
         
         self.language_codes = {
-            "English": "eng",
-            "Spanish": "spa",
-            "French": "fra",
-            "German": "deu",
-            "Italian": "ita",
-            "Portuguese": "por",
-            "Russian": "rus",
-            "Chinese": "cmn",
-            "Japanese": "jpn"
+            "English (US)": "eng",
+            "Spanish (ES)": "spa",
+            "French (FR)": "fra",
+            "German (DE)": "deu",
+            "Italian (IT)": "ita",
+            "Portuguese (BR)": "por",
+            "Russian (RU)": "rus",
+            "Chinese (CN)": "cmn",
+            "Japanese (JP)": "jpn",
+            "Korean (KR)": "kor",
+            "Hindi (IN)": "hin",
+            "Arabic (AR)": "ara"
         }
 
-    def translate(self, text: str, src_lang: str, tgt_lang: str) -> tuple[int, np.ndarray]:
+    def start_recording(self):
+        self.is_recording = True
+        threading.Thread(target=self._record_audio).start()
+
+    def stop_recording(self):
+        self.is_recording = False
+
+    def _record_audio(self):
+        with sd.InputStream(channels=1, samplerate=16000, callback=self._audio_callback):
+            while self.is_recording:
+                time.sleep(0.1)
+
+    def _audio_callback(self, indata, frames, time, status):
+        self.audio_queue.put(indata.copy())
+
+    def translate_realtime(self, audio_chunk, src_lang: str, tgt_lang: str) -> tuple[int, np.ndarray]:
+        try:
+            inputs = self.processor(audios=audio_chunk, return_tensors="pt")
+            audio_array = self.model.generate(**inputs, tgt_lang=self.language_codes[tgt_lang])[0].cpu().numpy().squeeze()
+            return self.sample_rate, audio_array
+        except Exception as e:
+            raise gr.Error(f"Translation failed: {str(e)}")
+
+    def translate_text(self, text: str, src_lang: str, tgt_lang: str) -> tuple[int, np.ndarray]:
         try:
             inputs = self.processor(text=text, src_lang=self.language_codes[src_lang], return_tensors="pt")
             audio_array = self.model.generate(**inputs, tgt_lang=self.language_codes[tgt_lang])[0].cpu().numpy().squeeze()
@@ -31,19 +63,52 @@ class SeamlessTranslator:
         except Exception as e:
             raise gr.Error(f"Translation failed: {str(e)}")
 
-# Custom CSS for Jarvis theme
 css = """
-#jarvis-interface {
-    background-color: black !important;
-    background-image: radial-gradient(circle at center, #00303030 0%, #00000080 100%);
+/* Cores e temas da interface */
+:root {
+    --primary: #00ffff;
+    --secondary: #0066cc;
+    --accent: #ff3366;
+    --background: #000000;
+    --text: #ffffff;
+}
+
+#aris-interface {
+    background-color: var(--background);
+    background-image: 
+        radial-gradient(circle at 20% 20%, rgba(0, 102, 204, 0.1) 0%, transparent 50%),
+        radial-gradient(circle at 80% 80%, rgba(0, 255, 255, 0.1) 0%, transparent 50%);
     min-height: 100vh;
     font-family: 'Courier New', monospace;
+    padding: 20px;
 }
 
+.title-container {
+    text-align: center;
+    color: var(--primary);
+    margin-bottom: 30px;
+    position: relative;
+}
+
+.title-container h1 {
+    font-size: 3em;
+    letter-spacing: 10px;
+    margin: 0;
+    text-shadow: 0 0 10px var(--primary);
+}
+
+.title-container h3 {
+    font-size: 1.2em;
+    letter-spacing: 3px;
+    opacity: 0.8;
+    margin: 5px 0;
+}
+
+/* Sistema de anéis central */
 #status-ring {
-    width: 300px;
-    height: 300px;
-    border: 4px solid #00ffff;
+    width: 400px;
+    height: 400px;
+    border: 4px solid var(--primary);
     border-radius: 50%;
     margin: 20px auto;
     position: relative;
@@ -51,174 +116,362 @@ css = """
     display: flex;
     align-items: center;
     justify-content: center;
+    background: 
+        radial-gradient(circle at center, rgba(0, 255, 255, 0.1) 0%, transparent 70%),
+        conic-gradient(from 0deg, transparent 0%, rgba(0, 255, 255, 0.1) 50%, transparent 100%);
 }
 
-@keyframes pulse {
-    0% { box-shadow: 0 0 0 0 rgba(0, 255, 255, 0.4); }
-    70% { box-shadow: 0 0 0 20px rgba(0, 255, 255, 0); }
-    100% { box-shadow: 0 0 0 0 rgba(0, 255, 255, 0); }
+#outer-ring-decoration {
+    position: absolute;
+    width: 420px;
+    height: 420px;
+    border-radius: 50%;
+    border: 1px solid rgba(0, 255, 255, 0.3);
+    animation: rotate 20s linear infinite;
 }
 
 #inner-ring {
-    width: 200px;
-    height: 200px;
-    border: 2px solid #00ffff;
+    width: 300px;
+    height: 300px;
+    border: 2px solid var(--primary);
     border-radius: 50%;
     display: flex;
     align-items: center;
     justify-content: center;
+    position: relative;
 }
 
 #core {
-    width: 100px;
-    height: 100px;
-    border: 3px solid #00ffff;
+    width: 200px;
+    height: 200px;
+    border: 3px solid var(--primary);
     border-radius: 50%;
-    background-color: black;
+    background-color: rgba(0, 0, 0, 0.8);
     display: flex;
+    flex-direction: column;
     align-items: center;
     justify-content: center;
-    color: #00ffff;
+    color: var(--primary);
     text-align: center;
-    padding: 10px;
+    padding: 15px;
+    position: relative;
+    box-shadow: 0 0 20px rgba(0, 255, 255, 0.2);
 }
 
-.jarvis-textbox {
-    background-color: black !important;
-    border: 2px solid #00ffff !important;
-    color: #00ffff !important;
+/* Animações */
+@keyframes pulse {
+    0% { box-shadow: 0 0 0 0 rgba(0, 255, 255, 0.4); }
+    70% { box-shadow: 0 0 0 20px rgba(0, 255, 255, 0); }
+    100% { box-shadow: 0 0 0 0 rgba(0, 255, 255, 0); }
+}
+
+@keyframes rotate {
+    from { transform: rotate(0deg); }
+    to { transform: rotate(360deg); }
+}
+
+/* Elementos da interface */
+.aris-controls {
+    background: rgba(0, 0, 0, 0.7);
+    border: 2px solid var(--primary);
+    border-radius: 10px;
+    padding: 20px;
+    margin: 20px 0;
+    box-shadow: 0 0 15px rgba(0, 255, 255, 0.1);
+}
+
+.aris-textbox {
+    background-color: rgba(0, 0, 0, 0.8) !important;
+    border: 2px solid var(--primary) !important;
+    color: var(--primary) !important;
     font-family: 'Courier New', monospace !important;
+    border-radius: 5px !important;
+    padding: 10px !important;
 }
 
-.jarvis-button {
+.aris-button {
     background-color: transparent !important;
-    border: 2px solid #00ffff !important;
-    color: #00ffff !important;
+    border: 2px solid var(--primary) !important;
+    color: var(--primary) !important;
     font-family: 'Courier New', monospace !important;
+    text-transform: uppercase !important;
+    letter-spacing: 2px !important;
+    padding: 12px 24px !important;
+    border-radius: 5px !important;
+    transition: all 0.3s ease !important;
 }
 
-.jarvis-button:hover {
+.aris-button:hover {
     background-color: rgba(0, 255, 255, 0.1) !important;
+    box-shadow: 0 0 15px rgba(0, 255, 255, 0.3) !important;
+    transform: translateY(-2px) !important;
 }
 
 .status-box {
-    background-color: black !important;
-    border: 2px solid #00ffff !important;
-    color: #00ffff !important;
-    padding: 10px !important;
+    background-color: rgba(0, 0, 0, 0.8) !important;
+    border: 2px solid var(--primary) !important;
+    color: var(--primary) !important;
+    padding: 15px !important;
     border-radius: 5px !important;
     margin: 5px !important;
     text-align: center !important;
+    text-transform: uppercase !important;
+    letter-spacing: 1px !important;
+    transition: all 0.3s ease !important;
+    position: relative;
+    overflow: hidden;
+}
+
+.status-box::before {
+    content: '';
+    position: absolute;
+    top: 0;
+    left: -100%;
+    width: 100%;
+    height: 2px;
+    background: linear-gradient(90deg, transparent, var(--primary));
+    animation: scan-line 2s linear infinite;
+}
+
+@keyframes scan-line {
+    0% { left: -100%; }
+    100% { left: 100%; }
+}
+
+.mode-indicator {
+    position: absolute;
+    top: 10px;
+    right: 10px;
+    padding: 5px 10px;
+    background-color: var(--accent);
+    color: var(--text);
+    border-radius: 3px;
+    font-size: 0.8em;
+    letter-spacing: 1px;
+}
+
+.stats-container {
+    display: grid;
+    grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
+    gap: 15px;
+    margin-top: 20px;
+}
+
+.stat-item {
+    background: rgba(0, 0, 0, 0.7);
+    border: 1px solid var(--primary);
+    padding: 10px;
+    border-radius: 5px;
+    text-align: center;
+    color: var(--primary);
+}
+
+.language-pair-display {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    gap: 10px;
+    margin: 10px 0;
+    color: var(--primary);
+    font-size: 1.2em;
+}
+
+.language-pair-display::before,
+.language-pair-display::after {
+    content: '⟨';
+    color: var(--secondary);
 }
 """
 
 def create_interface():
-    translator = SeamlessTranslator()
-    
-    def translate_text(text: str, src_lang: str, tgt_lang: str, progress=gr.Progress()):
-        progress(0, desc="Initializing...")
-        progress(0.3, desc="Processing text...")
-        sample_rate, audio = translator.translate(text, src_lang, tgt_lang)
-        progress(0.7, desc="Generating audio...")
-        progress(1.0, desc="Complete!")
-        return audio
+    translator = ARISTranslator()
     
-    with gr.Blocks(css=css, title="J.A.R.V.I.S Translator") as demo:
-        gr.Markdown(
-            """
-            # J.A.R.V.I.S TRANSLATION SYSTEM
-            ### Powered by SeamlessM4T
-            """
+    def update_status():
+        return (
+            f"A.R.I.S. CORE v2.0.0\n"
+            f"Time: {datetime.now().strftime('%H:%M:%S')}\n"
+            f"Neural Engine: ACTIVE\n"
+            f"Translation Matrix: OPERATIONAL"
         )
+
+    def start_realtime_translation(src_lang, tgt_lang):
+        translator.start_recording()
+        return "Real-time translation active..."
+
+    def stop_realtime_translation():
+        translator.stop_recording()
+        return "Translation stopped."
+    
+    with gr.Blocks(css=css, title="A.R.I.S. - Advanced Real-time Interpretation System") as demo:
+        gr.HTML('''
+            <div class="title-container">
+                <h1>A.R.I.S.</h1>
+                <h3>Advanced Real-time Interpretation System</h3>
+                <div class="mode-indicator">QUANTUM CORE ACTIVE</div>
+            </div>
+        ''')
         
-        # Jarvis interface container
-        with gr.Column(elem_id="jarvis-interface"):
-            # Status Ring
+        with gr.Column(elem_id="aris-interface"):
             gr.HTML("""
                 <div id="status-ring">
+                    <div id="outer-ring-decoration"></div>
                     <div id="inner-ring">
                         <div id="core">
-                            <div>JARVIS</div>
-                            <div>ACTIVE</div>
+                            <div>A.R.I.S.</div>
+                            <div>QUANTUM CORE</div>
+                            <div>v2.0.0</div>
+                            <div class="system-version">NEURAL ENGINE ACTIVE</div>
                         </div>
                     </div>
                 </div>
             """)
             
-            # Input controls
             with gr.Row():
-                text_input = gr.Textbox(
-                    label="Command Input",
-                    placeholder="Enter text to translate...",
-                    elem_classes=["jarvis-textbox"],
-                    lines=3
-                )
-            
-            with gr.Row():
-                src_lang = gr.Dropdown(
-                    choices=list(translator.language_codes.keys()),
-                    value="English",
-                    label="Source Language",
-                    elem_classes=["jarvis-textbox"]
-                )
-                tgt_lang = gr.Dropdown(
-                    choices=list(translator.language_codes.keys()),
-                    value="Spanish",
-                    label="Target Language",
-                    elem_classes=["jarvis-textbox"]
-                )
-            
-            translate_btn = gr.Button(
-                "▶ EXECUTE TRANSLATION",
-                elem_classes=["jarvis-button"]
-            )
-            
-            # Output audio
-            audio_output = gr.Audio(
-                label="Translated Speech",
-                type="numpy"
-            )
-            
-            # Status boxes
-            with gr.Row():
-                with gr.Column():
-                    gr.Markdown(
-                        """
-                        <div class="status-box">
-                            SYSTEM STATUS<br>
-                            <strong>ACTIVE</strong>
-                        </div>
-                        """
-                    )
                 with gr.Column():
-                    gr.Markdown(
-                        """
-                        <div class="status-box">
-                            AUDIO SYSTEM<br>
-                            <strong>READY</strong>
-                        </div>
-                        """
-                    )
+                    with gr.Tab("Real-time Translation"):
+                        src_lang_realtime = gr.Dropdown(
+                            choices=list(translator.language_codes.keys()),
+                            value="English (US)",
+                            label="SOURCE LANGUAGE",
+                            elem_classes=["aris-textbox"]
+                        )
+                        tgt_lang_realtime = gr.Dropdown(
+                            choices=list(translator.language_codes.keys()),
+                            value="Spanish (ES)",
+                            label="TARGET LANGUAGE",
+                            elem_classes=["aris-textbox"]
+                        )
+                        start_btn = gr.Button("▶ START REAL-TIME TRANSLATION", elem_classes=["aris-button"])
+                        stop_btn = gr.Button("⬛ STOP TRANSLATION", elem_classes=["aris-button"])
+                        status_realtime = gr.Textbox(label="REAL-TIME STATUS", elem_classes=["aris-textbox"])
+
+                    with gr.Tab("Text Translation"):
+                        text_input = gr.Textbox(
+                            label="INPUT TEXT",
+                            placeholder="Enter text for translation...",
+                            elem_classes=["aris-textbox"],
+                            lines=3
+                        )
+                        with gr.Row():
+                            src_lang_text = gr.Dropdown(
+                                choices=list(translator.language_codes.keys()),
+                                value="English (US)",
+                                label="SOURCE LANGUAGE",
+                                elem_classes=["aris-textbox"]
+                            )
+                            tgt_lang_text = gr.Dropdown(
+                                choices=list(translator.language_codes.keys()),
+                                value="Spanish (ES)",
+                                label="TARGET LANGUAGE",
+                                elem_classes=["aris-textbox"]
+                            )
+                        translate_btn = gr.Button("▶ TRANSLATE TEXT", elem_classes=["aris-button"])
+                
                 with gr.Column():
-                    gr.Markdown(
-                        """
-                        <div class="status-box">
-                            TRANSLATION<br>
-                            <strong>ONLINE</strong>
-                        </div>
-                        """
+                    audio_output = gr.Audio(
+                        label="TRANSLATION OUTPUT",
+                        type="numpy"
                     )
+                    
+                    with gr.Row():
+                        with gr.Column(min_width=200):
+                            gr.HTML(
+                                """
+                                <div class="status-box">
+                                    NEURAL CORE<br>
+                                    <strong>OPERATIONAL</strong>
+                                </div>
+                                """
+                            )
+                        with gr.Column(min_width=200):
+                            gr.HTML(
+                                """
+                                <div class="status-box">
+                                    QUANTUM ENGINE<br>
+                                    <strong>ACTIVE</strong>
+                                </div>
+                                """
+                            )
+                    
+                    with gr.Row():
+                        with gr.Column(min_width=200):
+                            gr.HTML(
+                                """
+                                <div class="status-box">
+                                    TRANSLATION MATRIX<br>
+                                    <strong>CALIBRATED</strong>
+                                </div>
+                                """
+                            )
+                        with gr.Column(min_width=200):
+                            gr.HTML(
+                                """
+                                <div class="status-box">
+                                    VOICE SYNTHESIS<br>
+                                    <strong>READY</strong>
+                                </div>
+                                """
+                            )
+                    
+                    # Sistema de estatísticas
+                    with gr.Row():
+                        gr.HTML("""
+                            <div class="stats-container">
+                                <div class="stat-item">
+                                    <div>Processing Speed</div>
+                                    <div style="font-size: 1.2em; margin: 5px 0;">0.8ms</div>
+                                </div>
+                                <div class="stat-item">
+                                    <div>Neural Load</div>
+                                    <div style="font-size: 1.2em; margin: 5px 0;">78%</div>
+                                </div>
+                                <div class="stat-item">
+                                    <div>Memory Usage</div>
+                                    <div style="font-size: 1.2em; margin: 5px 0;">4.2GB</div>
+                                </div>
+                            </div>
+                        """)
+
+            # Event handlers
+            def update_stats():
+                return {
+                    status_realtime: f"System Status: Active\nMemory Usage: {np.random.randint(70, 90)}%\nProcessing Speed: {np.random.randint(1, 5)}ms"
+                }
+
+            start_btn.click(
+                fn=start_realtime_translation,
+                inputs=[src_lang_realtime, tgt_lang_realtime],
+                outputs=status_realtime
+            )
             
-            # Event handler
+            stop_btn.click(
+                fn=stop_realtime_translation,
+                outputs=status_realtime
+            )
+
             translate_btn.click(
-                fn=translate_text,
-                inputs=[text_input, src_lang, tgt_lang],
+                fn=translator.translate_text,
+                inputs=[text_input, src_lang_text, tgt_lang_text],
                 outputs=audio_output
             )
-    
+
+            # Atualizações automáticas
+            demo.load(fn=update_status, outputs=status_realtime)
+            
     return demo
 
 if __name__ == "__main__":
     demo = create_interface()
     demo.queue()
-    demo.launch()
+    demo.launch()
+
+# Arquivo requirements.txt atualizado
+"""
+gradio>=4.0.0
+torch>=2.0.0
+torchaudio>=2.0.0
+transformers
+sentencepiece>=0.1.99
+numpy>=1.21.0
+sounddevice>=0.4.5
+"""