ai: Introduce new API of J.A.R.V.I.S.

Support streaming, real-time response mode.

assets/bin/ai

@@ -5,38 +5,121 @@
 #
 
 import sys
+import re
 
 from gradio_client import Client
-from rich.console import Console
+from rich.console import Console, Group
 from rich.markdown import Markdown
+from rich.syntax import Syntax
+from rich.live import Live
 
+# Prepares the display screen to show the final output
 console = Console()
+
+# Creates a connection to the server
 jarvis = Client("hadadrjt/ai")
 
-model = "JARVIS: 2.1.2" # You can change the models here.
+# Defines the specific AI model to use for responding
+# Change to the model you want, see:
+# https://huggingface.co/spaces/hadadrjt/ai/blob/main/docs/API.md#multi-platform
+model = "JARVIS: 2.1.3"
 
+# Reads user-provided input from the command line, if available
 args = sys.argv[1:]
+
+# Keeps track of whether deep search mode is activated
 deep_search = False
+
+# Checks if the input includes "-d", which turns on deep search
 if "-d" in args:
     deep_search = True
     args.remove("-d")
-input_text = " ".join(args) if args else "Hi!"
+
+# Combines the rest of the input into one complete message
+# If nothing was typed, it defaults to a basic greeting
+input = " ".join(args) if args else "Hi!"
+
+# Sets the AI to the desired version before sending any messages
 jarvis.predict(new=model, api_name="/change_model")
-if deep_search and model != "JARVIS: 2.1.2": # DO NOT CHANGE THIS!
-    console.print("")
-    console.print("------------------------------------------------------")
-    console.print("[bold red]Deep Search is only available for model JARVIS: 2.1.2.[/bold red]")
-    console.print("------------------------------------------------------")
-    console.print("")
+
+# Ensures deep search is only enabled for the correct AI model
+if deep_search and model != "JARVIS: 2.1.3":
     deep_search = False
+
+# Prepares and structures the AI’s response for display
+def layout(text):
+    # Searches for blocks of code within the full response text
+    code_blocks = list(re.finditer(r"\n\n```(.*?)\n\n(.*?)\n\n```\n\n\n", text, re.DOTALL))
+    segments = []  # Stores parts of the final display, including both text and code
+    last_end = 0  # Tracks where the previous segment ended
+
+    # Loops through each code block found in the AI's response
+    for block in code_blocks:
+        # Collects any normal text before the current code block
+        pre_text = text[last_end:block.start()]
+        if pre_text.strip():
+            # Converts plain text into styled text for easier reading
+            segments.append(Markdown(prepare_markdown(pre_text.strip())))
+
+        # Identifies the programming language used in the code block, if available
+        lang = block.group(1).strip() or "text"
+
+        # Extracts the actual code content
+        code = block.group(2).rstrip()
+
+        # Formats the code with syntax highlighting for clear presentation
+        segments.append(Syntax(code, lang, theme="monokai", line_numbers=False, word_wrap=True))
+
+        # Updates the position tracker to move past the current code block
+        last_end = block.end()
+
+    # Checks for any remaining text after the last code block
+    remaining = text[last_end:]
+    if remaining.strip():
+        # Formats and adds this final portion of text to the display
+        segments.append(Markdown(prepare_markdown(remaining.strip())))
+
+    # Returns a complete set of styled segments ready to be shown
+    return Group(*segments)
+
+# Adjusts special characters in the text to ensure they display correctly
+def prepare_markdown(text):
+    return text.replace("•", "*")
+
+# Displays the AI's response in real time as it’s being received
+def response(jarvis):
+    buffer = ""  # Holds the entire reply as it builds up
+    with Live(console=console, transient=False) as live:
+        # Continuously receives and processes parts of the reply
+        for partial in jarvis:
+            # Extracts the latest version of the message from the AI
+            text = partial[0][0][1]
+
+            # Determines what has changed since the last update
+            if text.startswith(buffer):
+                delta = text[len(buffer):]
+            else:
+                delta = text
+
+            # Updates the full message with any new content
+            buffer = text
+
+            # Refreshes the screen with the most recent version of the reply
+            live.update(layout(buffer))
+
+    # Ensures the final reply is printed once it’s complete
+    console.print()
+
+# Sends the user's input to the AI and selects the appropriate method
 if deep_search:
-    result = jarvis.predict(
-        multi={"text": input_text},
-        deep_search=True,
-        api_name="/respond_async")
+    # Uses the deeper, slower method for more thoughtful responses
+    jarvis = jarvis.submit(multi={"text": input}, deep_search=True, api_name="/respond_async")
 else:
-    result = jarvis.predict(
-        multi={"text": input_text},
-        api_name="/api")
-response_text = result[0][0][1]
-console.print(Markdown(response_text))
+    # Uses the standard method for quicker, more direct replies
+    jarvis = jarvis.submit(multi={"text": input}, api_name="/api")
+
+# Create a line break before the main AI's response
+print("")
+
+# Begins the process of showing the AI's response on the screen
+response(jarvis)

assets/bin/install.sh

@@ -0,0 +1,36 @@
+#!/bin/sh
+#
+# SPDX-FileCopyrightText: Hadad <[email protected]>
+# SPDX-License-Identifier: Apache-2.0
+#
+
+echo "Installing required Python packages..."
+pip install gradio_client rich --upgrade
+echo "Installation complete."
+echo ""
+echo ""
+echo "Downloading the J.A.R.V.I.S. script..."
+wget https://huggingface.co/spaces/hadadrjt/ai/raw/main/assets/bin/ai
+echo "Download complete."
+echo ""
+echo ""
+echo "Setting executable permission..."
+chmod a+x ai
+echo "Permission set."
+echo ""
+echo ""
+echo "Removing installer script..."
+rm install.sh
+echo "Done."
+echo ""
+echo ""
+echo "To send a regular message:"
+echo "./ai Your message here"
+echo ""
+echo "To use Deep Search mode:"
+echo "./ai -d Your message here"
+echo ""
+echo ""
+echo "For more details and advanced options, visit:"
+echo "https://huggingface.co/spaces/hadadrjt/ai/blob/main/docs/API.md#installations"
+echo ""

docs/API.md

@@ -4,25 +4,16 @@
 # Make sure you have "wget", "python3" and "pip" installed.
 # This package have very small size.
 
-pip install gradio_client rich --upgrade
-```
-
-#### DOWNLOAD JARVIS SCRIPT
-```bash
-# Terminal script.
-wget https://huggingface.co/spaces/hadadrjt/ai/raw/main/assets/bin/ai
-
-# Set permission.
-chmod a+x ai
+wget https://huggingface.co/spaces/hadadrjt/ai/raw/main/assets/bin/install.sh && chmod a+x install.sh && ./install.sh
 ```
 
 ### RUN JARVIS IN YOUR TERMINAL
 ```bash
 # Example normal usage.
-./ai "Your message here."
+./ai Your message here.
 
 # Example with Deep Search.
-./ai -d "Your message here."
+./ai -d Your message here.
 ```
 
 #### LINUX USER's
@@ -31,13 +22,13 @@ chmod a+x ai
 sudo mv ai /bin/
 
 # Now you can run with simple command.
-ai "Your message here."
+ai Your message here.
 ```
 
 ### MULTI PLATFORM
 ```
 Edit JARVIS terminal script, and find this code:
-model = "JARVIS: 2.1.2"
+model = "JARVIS: 2.1.3"
 
 Choose one of the model name for the JARVIS multi platform.
 
@@ -66,5 +57,5 @@ Choose one of the model name for the JARVIS multi platform.
 Example:
 model = "DeepSeek: V3-0324"
 
-Please note that the Deep Search feature is only available for JARVIS: 2.1.2.
+Please note that the Deep Search feature is only available for JARVIS: 2.1.3.
 ```