Update app.py

app.py

@@ -1,207 +1,65 @@
-import logging
-
-from h2o_wave import Q, main, app, copy_expando, handle_on, on
-import whisper
-
-import cards
-from utils import get_inline_script
-
-# Set up logging
-logging.basicConfig(format='%(levelname)s:\t[%(asctime)s]\t%(message)s', level=logging.INFO)
-
-
-@app('/')
+from h2o_wave import main, app, Q, ui, copy_expando
+from transformers import pipeline
+
+async def init(q: Q):
+    if not q.client.app_initialized:
+        q.app.model = pipeline("text-generation")
+        q.client.app_initialized = True
+
+    q.page.drop()
+
+    q.page["title"] = ui.header_card(
+        box="1 1 8 1",
+        title="Text Generation",
+        subtitle="Generate text using Huggingface pipelines",
+        icon="AddNotes",
+        icon_color="Blue",
+    )
+
+async def get_inputs(q: Q):
+    q.page['main'] = ui.form_card(box="1 2 8 5", items=[
+        ui.text_xl('Enter your text input for generation:'),
+        ui.textbox(name="input_text",
+                   label='',
+                   value=q.app.input_text,
+                   multiline=True),
+        ui.separator(),
+        ui.slider(name="num_words_to_generate",
+                  label="Maximum number of words to generate (including input text)",
+                  min=5,
+                  max=50,
+                  step=1,
+                  value=q.app.num_words_to_generate if q.app.num_words_to_generate else 12,
+                  ),
+        ui.separator(),
+        ui.buttons([ui.button(name="generate_text", label='Generate', primary=True),
+                    ])
+    ])
+
+async def show_results(q: Q):
+    q.page['main'] = ui.form_card(box="1 2 4 5", items=[
+        ui.text_xl("Input Text:"),
+        ui.separator(),
+        ui.text(q.app.input_text),
+        ui.separator(),
+        ui.buttons([ui.button(name="get_inputs", label='Try Again!', primary=True),
+                    ])
+    ])
+
+    result = q.app.model(q.app.input_text, max_length=q.app.num_words_to_generate, do_sample=False)[0]
+    q.app.generated_text = result["generated_text"]
+    q.page['visualization'] = ui.form_card(box="5 2 4 5", items=[
+        ui.text_xl("Generated Text:"),
+        ui.separator(''),
+        ui.text(q.app.generated_text)
+    ])
+
+@app("/")
 async def serve(q: Q):
-    """
-    Main entry point. All queries pass through this function.
-    """
-
-    try:
-        # Initialize the app if not already
-        if not q.app.initialized:
-            await initialize_app(q)
-
-        # Initialize the client if not already
-        if not q.client.initialized:
-            await initialize_client(q)
-
-        # Update theme if toggled
-        elif q.args.theme_dark is not None and q.args.theme_dark != q.client.theme_dark:
-            await update_theme(q)
-
-        # Run inference if audio is recorded
-        elif q.events.audio:
-            await audio_inference(q)
-
-        # Delegate query to query handlers
-        elif await handle_on(q):
-            pass
-
-        # Adding this condition to help in identifying bugs
-        else:
-            await handle_fallback(q)
-
-    except Exception as error:
-        await show_error(q, error=str(error))
-
-
-async def initialize_app(q: Q):
-    """
-    Initialize the app.
-    """
-
-    logging.info('Initializing app')
-
-    # Set initial argument values
-    q.app.cards = ['main', 'error']
-
-    q.app.model = whisper.load_model('base')
-
-    q.app.initialized = True
-
-
-async def initialize_client(q: Q):
-    """
-    Initialize the client (browser tab).
-    """
-
-    logging.info('Initializing client')
-
-    # Set initial argument values
-    q.client.theme_dark = True
-
-    # Add layouts, scripts, header and footer
-    q.page['meta'] = cards.meta
-    q.page['header'] = cards.header
-    q.page['footer'] = cards.footer
-
-    # Add cards for the main page
-    q.page['asr'] = cards.asr()
-
-    q.client.initialized = True
-
-    await q.page.save()
-
-
-async def update_theme(q: Q):
-    """
-    Update theme of app.
-    """
-
-    # Copying argument values to client
-    copy_expando(q.args, q.client)
-
-    if q.client.theme_dark:
-        logging.info('Updating theme to dark mode')
-
-        # Update theme from light to dark mode
-        q.page['meta'].theme = 'h2o-dark'
-        q.page['header'].icon_color = 'black'
+    await init(q)
+    if q.args.generate_text:
+        copy_expando(q.args, q.app)
+        await show_results(q)
     else:
-        logging.info('Updating theme to light mode')
-
-        # Update theme from dark to light mode
-        q.page['meta'].theme = 'light'
-        q.page['header'].icon_color = '#FEC924'
-
-    await q.page.save()
-
-
-@on('start')
-async def start_recording(q: Q):
-    """
-    Start recording audio.
-    """
-
-    logging.info('Starting recording')
-
-    q.page['meta'].script = get_inline_script('startRecording()')
-    q.page['asr'] = cards.asr(recording=True)
-
-    await q.page.save()
-
-
-@on('stop')
-async def stop_recording(q: Q):
-    """
-    Stop recording audio.
-    """
-
-    logging.info('Stopping recording')
-
-    q.page['meta'].script = get_inline_script('stopRecording()')
-    q.page['asr'] = cards.asr()
-
-    await q.page.save()
-
-
-@on('audio')
-async def audio_inference(q: Q):
-    """
-    Running ASR inference on audio.
-    """
-
-    logging.info('Inferencing recorded audio')
-
-    audio_path = await q.site.download(q.events.audio.captured, '.')
-
-    q.client.transcription = q.app.model.transcribe(audio_path)['text']
-
-    q.page['asr'] = cards.asr(audio_path=q.events.audio.captured, transcription=q.client.transcription)
-
-    await q.page.save()
-
-
-def clear_cards(q: Q, card_names: list):
-    """
-    Clear cards from the page.
-    """
-
-    logging.info('Clearing cards')
-
-    # Delete cards from the page
-    for card_name in card_names:
-        del q.page[card_name]
-
-
-async def show_error(q: Q, error: str):
-    """
-    Displays errors.
-    """
-
-    logging.error(error)
-
-    # Clear all cards
-    clear_cards(q, q.app.cards)
-
-    # Format and display the error
-    q.page['error'] = cards.crash_report(q)
-
-    await q.page.save()
-
-
-@on('reload')
-async def reload_client(q: Q):
-    """
-    Reset the client.
-    """
-
-    logging.info('Reloading client')
-
-    # Clear all cards
-    clear_cards(q, q.app.cards)
-
-    # Reload the client
-    await initialize_client(q)
-
-
-async def handle_fallback(q: Q):
-    """
-    Handle fallback cases.
-    """
-
-    logging.info('Adding fallback page')
-
-    q.page['fallback'] = cards.fallback
-
-    await q.page.save()
+        await get_inputs(q)
+    await q.page.save()