ngpt 2.16.5__tar.gz → 2.16.6__tar.gz

{ngpt-2.16.5 → ngpt-2.16.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 2.16.5
+Version: 2.16.6
 Summary: Swiss army knife for LLMs: powerful CLI, interactive chatbot, and flexible Python library. Works with OpenAI, Ollama, Groq, Claude, and any OpenAI-compatible API.
 Project-URL: Homepage, https://github.com/nazdridoy/ngpt
 Project-URL: Repository, https://github.com/nazdridoy/ngpt

{ngpt-2.16.5 → ngpt-2.16.6}/docs/api/cli.md

@@ -160,7 +160,7 @@ from ngpt.cli.renderers import prettify_streaming_markdown
 def prettify_streaming_markdown(renderer='rich', is_interactive=False, header_text=None)
 ```
 
-Creates a streaming markdown renderer that updates in real-time.
+Creates a streaming markdown renderer that updates in real-time with loading spinner functionality.
 
 **Parameters:**
 - `renderer` (str): Which renderer to use ('auto', 'rich', 'glow')
@@ -168,21 +168,47 @@ Creates a streaming markdown renderer that updates in real-time.
 - `header_text` (str, optional): Header text to display above the content
 
 **Returns:**
-- object: A streaming markdown renderer object with an `update_content` method
+- tuple: (live_display, update_function, setup_spinner_func) if successful, (None, None, None) otherwise
+  - `live_display`: The rich.Live display object for controlling the display lifecycle
+  - `update_function`: Function to call with updated content that will refresh the display
+  - `setup_spinner_func`: Function to set up a spinner while waiting for first content
 
 **Example:**
 ```python
 from ngpt import NGPTClient, load_config
 from ngpt.cli.renderers import prettify_streaming_markdown
+import threading
 
 client = NGPTClient(**load_config())
-streamer = prettify_streaming_markdown(renderer='rich')
 
-client.chat(
-    "Explain quantum computing with code examples",
+# Get components for streaming display with spinner
+live_display, update_function, setup_spinner = prettify_streaming_markdown(renderer='rich')
+
+# Set up spinner (optional)
+stop_spinner_event = threading.Event()
+stop_spinner_func = None
+if setup_spinner:
+    stop_spinner_func = setup_spinner(stop_spinner_event, "Waiting for response...")
+
+# The update_function will automatically:
+# 1. Start the live display when first content arrives
+# 2. Stop the spinner when first content arrives
+# 3. Update the display with new content
+
+# Use with client
+response = client.chat(
+    "Explain quantum computing",
     stream=True,
-    stream_callback=streamer.update_content
+    stream_callback=update_function
 )
+
+# Ensure spinner is stopped if no content was received
+if not stop_spinner_event.is_set():
+    stop_spinner_event.set()
+
+# Stop the display when done
+if live_display:
+    live_display.stop()
 ```
 
 ### `has_markdown_renderer`

{ngpt-2.16.5 → ngpt-2.16.6}/docs/examples/advanced.md

@@ -217,38 +217,33 @@ def custom_streaming_markdown():
     # Create a Rich console
     console = Console()
     
-    # Initialize the markdown renderer
-    markdown_renderer = prettify_streaming_markdown(renderer='rich')
+    # Initialize a real-time markdown renderer
+    live_display, update_function, setup_spinner = prettify_streaming_markdown(renderer='rich')
     
-    # Start a conversation
-    console.print("[bold green]Starting conversation with markdown rendering[/bold green]")
-    console.print("[bold]Type 'exit' to quit[/bold]")
-    console.print("-" * 50)
+    # Set up spinner for waiting period
+    import threading
+    stop_spinner_event = threading.Event()
+    if setup_spinner:
+        stop_spinner_func = setup_spinner(stop_spinner_event, "Waiting for response...")
     
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant with markdown capabilities. Use formatting like headers, lists, code blocks, and tables to present information clearly."}
-    ]
+    # Start the live display
+    markdown_renderer.start()
     
-    while True:
-        user_input = input("You: ")
-        if user_input.lower() in ['exit', 'quit', 'bye']:
-            break
-        
-        # Add user message to history
-        messages.append({"role": "user", "content": user_input})
-        
-        # Call the API with the custom renderer
-        console.print("[bold blue]Assistant:[/bold blue]")
-        response = client.chat(
-            "", 
-            messages=messages, 
-            stream=True, 
-            markdown_format=True,
-            stream_callback=markdown_renderer.update_content
-        )
+    # Stream the response with real-time rendering
+    for chunk in client.chat(
+        "Explain quantum computing briefly",
+        stream=True
+    ):
+        full_response += chunk
+        update_function(full_response)
+    
+    # Ensure spinner is stopped if still running
+    if not stop_spinner_event.is_set():
+        stop_spinner_event.set()
         
-        # Add the response to history
-        messages.append({"role": "assistant", "content": response})
+    # Stop the live display when done
+    if live_display:
+        live_display.stop()
 
 if __name__ == "__main__":
     custom_streaming_markdown()
@@ -584,101 +579,114 @@ def main():
                 log_file=log_file
             )
             
-        # Standard chat mode
-        else:
-            # Real-time prettified markdown mode
-            if args.stream_prettify:
-                streamer = prettify_streaming_markdown(renderer='rich')
+        # Real-time prettified markdown mode
+        if args.stream_prettify:
+            live_display, update_function, setup_spinner = prettify_streaming_markdown(renderer='rich')
+            
+            # Set up spinner for waiting period
+            import threading
+            stop_spinner_event = threading.Event()
+            if setup_spinner:
+                stop_spinner_func = setup_spinner(stop_spinner_event, "Waiting for response...")
+            
+            response = client.chat(
+                prompt, 
+                temperature=args.temperature,
+                stream=True,
+                stream_callback=update_function,
+                markdown_format=True,
+                web_search=args.web_search,
+                messages=messages
+            )
+            
+            # Ensure spinner is stopped if still running
+            if not stop_spinner_event.is_set():
+                stop_spinner_event.set()
+            
+            # Stop the live display when done
+            if live_display:
+                live_display.stop()
+            
+            if args.output:
+                with open(args.output, 'w') as f:
+                    f.write(response)
+                print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
+                
+            if log_file:
+                log_file.write(f"Assistant: {response}\n\n")
+            
+        # Basic prettify mode
+        elif args.prettify:
+            from rich.markdown import Markdown
+            from rich.console import Console
+            
+            console = Console()
+            
+            if args.output:
+                # No streaming if saving to file
                 response = client.chat(
                     prompt, 
-                    temperature=args.temperature,
-                    stream=True,
-                    stream_callback=streamer.update_content,
-                    markdown_format=True,
+                    temperature=args.temperature, 
+                    stream=False,
                     web_search=args.web_search,
+                    markdown_format=True,
                     messages=messages
                 )
+                with open(args.output, 'w') as f:
+                    f.write(response)
+                print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
                 
-                if args.output:
-                    with open(args.output, 'w') as f:
-                        f.write(response)
-                    print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
-                    
                 if log_file:
                     log_file.write(f"Assistant: {response}\n\n")
-                    
-            # Basic prettify mode
-            elif args.prettify:
-                from rich.markdown import Markdown
-                from rich.console import Console
+            else:
+                # Use rich to render markdown after completion
+                response = client.chat(
+                    prompt, 
+                    temperature=args.temperature, 
+                    stream=False,
+                    web_search=args.web_search,
+                    markdown_format=True,
+                    messages=messages
+                )
+                console.print(Markdown(response))
                 
-                console = Console()
+                if log_file:
+                    log_file.write(f"Assistant: {response}\n\n")
+            
+        # Simple mode
+        else:
+            if args.output:
+                # No streaming if saving to file
+                response = client.chat(
+                    prompt, 
+                    temperature=args.temperature, 
+                    stream=False,
+                    web_search=args.web_search,
+                    messages=messages
+                )
+                with open(args.output, 'w') as f:
+                    f.write(response)
+                print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
                 
-                if args.output:
-                    # No streaming if saving to file
-                    response = client.chat(
-                        prompt, 
-                        temperature=args.temperature, 
-                        stream=False,
-                        web_search=args.web_search,
-                        markdown_format=True,
-                        messages=messages
-                    )
-                    with open(args.output, 'w') as f:
-                        f.write(response)
-                    print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
-                    
-                    if log_file:
-                        log_file.write(f"Assistant: {response}\n\n")
-                else:
-                    # Use rich to render markdown after completion
-                    response = client.chat(
-                        prompt, 
-                        temperature=args.temperature, 
-                        stream=False,
-                        web_search=args.web_search,
-                        markdown_format=True,
-                        messages=messages
-                    )
-                    console.print(Markdown(response))
-                    
-                    if log_file:
-                        log_file.write(f"Assistant: {response}\n\n")
-                        
-            # Simple mode
+                if log_file:
+                    log_file.write(f"Assistant: {response}\n\n")
             else:
-                if args.output:
-                    # No streaming if saving to file
-                    response = client.chat(
-                        prompt, 
-                        temperature=args.temperature, 
-                        stream=False,
-                        web_search=args.web_search,
-                        messages=messages
-                    )
-                    with open(args.output, 'w') as f:
-                        f.write(response)
-                    print(f"{COLORS['green']}Response saved to {args.output}{COLORS['reset']}")
-                    
-                    if log_file:
-                        log_file.write(f"Assistant: {response}\n\n")
-                else:
-                    # Stream to console
-                    full_response = ""
-                    for chunk in client.chat(
-                        prompt, 
-                        temperature=args.temperature, 
-                        stream=True,
-                        web_search=args.web_search,
-                        messages=messages
-                    ):
-                        print(chunk, end="", flush=True)
-                        full_response += chunk
-                    print()  # Final newline
-                    
-                    if log_file:
-                        log_file.write(f"Assistant: {full_response}\n\n")
+                # Stream to console
+                full_response = ""
+                for chunk in client.chat(
+                    prompt, 
+                    temperature=args.temperature, 
+                    stream=True,
+                    web_search=args.web_search,
+                    messages=messages
+                ):
+                    print(chunk, end="", flush=True)
+                    full_response += chunk
+                print()  # Final newline
                 
+                if log_file:
+                    log_file.write(f"Assistant: {full_response}\n\n")
+        
         if log_file:
             log_file.close()
             

{ngpt-2.16.5 → ngpt-2.16.6}/docs/examples/cli_components.md

@@ -260,17 +260,35 @@ CODE:
     # Use streaming markdown renderer if Rich is available
     if has_rich and has_markdown_renderer(renderer='rich'):
         # Create a streaming markdown renderer
-        markdown_streamer = prettify_streaming_markdown(
+        live_display, update_function, setup_spinner = prettify_streaming_markdown(
             renderer='rich',
             header_text=f"Documentation for {args.file}"
         )
         
+        # Setup spinner for waiting period
+        import threading
+        stop_spinner_event = threading.Event()
+        stop_spinner_func = None
+        if setup_spinner:
+            stop_spinner_func = setup_spinner(stop_spinner_event, "Generating documentation...")
+        
         # Stream the response with live updating
         try:
             for chunk in client.chat(prompt, stream=True):
                 full_response += chunk
-                markdown_streamer.update_content(full_response)
+                update_function(full_response)
+            
+            # Ensure spinner is stopped if still running
+            if not stop_spinner_event.is_set():
+                stop_spinner_event.set()
+            
+            # Stop the display when done
+            if live_display:
+                live_display.stop()
         except Exception as e:
+            # Ensure spinner is stopped on error
+            if not stop_spinner_event.is_set():
+                stop_spinner_event.set()
             print(f"\nError generating documentation: {e}", file=sys.stderr)
             sys.exit(1)
     else:
@@ -511,11 +529,18 @@ Return ONLY the improved text without explanations or notes."""
         if args.stream:
             # Stream with live updates
             if args.prettify:
-                streamer = prettify_streaming_markdown(
+                live_display, update_function, setup_spinner = prettify_streaming_markdown(
                     renderer='rich',
                     header_text="Improved Text"
                 )
                 
+                # Setup spinner for waiting period
+                import threading
+                stop_spinner_event = threading.Event()
+                stop_spinner_func = None
+                if setup_spinner:
+                    stop_spinner_func = setup_spinner(stop_spinner_event, "Improving text...")
+                
                 full_response = ""
                 for chunk in client.chat(
                     text,
@@ -523,7 +548,15 @@ Return ONLY the improved text without explanations or notes."""
                     stream=True
                 ):
                     full_response += chunk
-                    streamer.update_content(full_response)
+                    update_function(full_response)
+                
+                # Ensure spinner is stopped if still running
+                if not stop_spinner_event.is_set():
+                    stop_spinner_event.set()
+                
+                # Stop the display when done
+                if live_display:
+                    live_display.stop()
                 
                 improved_text = full_response
             else:

{ngpt-2.16.5 → ngpt-2.16.6}/docs/usage/library_usage.md

@@ -381,37 +381,51 @@ config = load_config()
 client = NGPTClient(**config)
 
 # Get the markdown streamer
-markdown_streamer = prettify_streaming_markdown(
+live_display, update_function, setup_spinner = prettify_streaming_markdown(
     renderer='rich',
     header_text="Streaming Response:"
 )
 
-# Start the live display
-if markdown_streamer[0]:  # Check if setup was successful
-    markdown_streamer[0].start()
+# Set up spinner while waiting for first content
+if live_display and setup_spinner:
+    import threading
+    stop_spinner_event = threading.Event()
+    stop_spinner_func = setup_spinner(stop_spinner_event, "Waiting for response...")
+    
+    # The live display will start automatically when the first content is received
+    # (no need to call live_display.start() manually)
 
-    # Use the streamer with the client
+    # Use the update function with the client
     response = client.chat(
         "Explain quantum computing with code examples",
         stream=True,
         markdown_format=True,
-        stream_callback=markdown_streamer[1]
+        stream_callback=update_function
     )
     
+    # Ensure spinner is stopped if no content was received
+    if not stop_spinner_event.is_set():
+        stop_spinner_event.set()
+    
     # Stop the live display when done
-    markdown_streamer[0].stop()
+    live_display.stop()
 ```
 
 For more control, you can access the live display and update function directly:
 
 ```python
-live_display, update_function = prettify_streaming_markdown(
+live_display, update_function, setup_spinner = prettify_streaming_markdown(
     renderer='rich',
     header_text="Custom Header"
 )
 
 if live_display:  # Check if setup was successful
-    live_display.start()
+    # Optional: Set up a spinner while waiting for content
+    import threading
+    stop_spinner_event = threading.Event()
+    stop_spinner_func = setup_spinner(stop_spinner_event, "Processing...")
+    
+    # The first update_function call will automatically start the display and stop the spinner
     
     # Update the content manually
     update_function("# Header\nThis is *formatted* content")

{ngpt-2.16.5 → ngpt-2.16.6}/ngpt/cli/modes/rewrite.py

@@ -131,8 +131,6 @@ def rewrite_mode(client, args, logger=None):
         {"role": "user", "content": input_text}
     ]
     
-    print("\nSubmission successful. Waiting for response...")
-    
     # Log the messages if logging is enabled
     if logger:
         logger.log("system", REWRITE_SYSTEM_PROMPT)

{ngpt-2.16.5 → ngpt-2.16.6}/ngpt/cli/modes/text.py

@@ -21,8 +21,6 @@ def text_mode(client, args, logger=None):
             print("Exiting.")
             return
     
-    print("\nSubmission successful. Waiting for response...")
-    
     # Log the user message if logging is enabled
     if logger:
         logger.log("user", prompt)

{ngpt-2.16.5 → ngpt-2.16.6}/ngpt/cli/renderers.py

@@ -239,8 +239,14 @@ def prettify_streaming_markdown(renderer='rich', is_interactive=False, header_te
         else:
             md_obj = Markdown("")
         
-        # Initialize the Live display with an empty markdown
-        live = Live(md_obj, console=console, refresh_per_second=10, auto_refresh=False)
+        # Initialize the Live display with vertical overflow handling
+        live = Live(
+            md_obj, 
+            console=console, 
+            refresh_per_second=10, 
+            auto_refresh=False,
+            vertical_overflow="visible"  # Attempt to make overflow visible
+        )
         
         # Track if this is the first content update
         first_update = True
@@ -259,7 +265,7 @@ def prettify_streaming_markdown(renderer='rich', is_interactive=False, header_te
                 sys.stdout.flush()
                 live.start()
             
-            # Update content
+            # Update content in live display
             if is_interactive and header_text:
                 # Update the panel content
                 md_obj.renderable = Markdown(content)

{ngpt-2.16.5 → ngpt-2.16.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ngpt"
-version = "2.16.5"
+version = "2.16.6"
 description = "Swiss army knife for LLMs: powerful CLI, interactive chatbot, and flexible Python library. Works with OpenAI, Ollama, Groq, Claude, and any OpenAI-compatible API."
 authors = [
     {name = "nazDridoy", email = "nazdridoy399@gmail.com"},

{ngpt-2.16.5 → ngpt-2.16.6}/uv.lock

@@ -134,7 +134,7 @@ wheels = [
 
 [[package]]
 name = "ngpt"
-version = "2.16.5"
+version = "2.16.6"
 source = { editable = "." }
 dependencies = [
     { name = "prompt-toolkit" },