Skip to content

Instantly share code, notes, and snippets.

@saifsmailbox98

saifsmailbox98/main.py

Created July 26, 2025 12:17
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save saifsmailbox98/08d56d318db535cd17783b401ce91775 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save saifsmailbox98/08d56d318db535cd17783b401ce91775 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. saifsmailbox98 created this gist Jul 26, 2025.
    69 changes: 69 additions & 0 deletions main.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,69 @@
    from os import getenv
    from dotenv import load_dotenv
    from rich.console import Console
    from rich.live import Live
    from rich.panel import Panel

    from perplexity_webui_scraper import Perplexity, CitationMode, ModelType, SearchFocus, SourceFocus, TimeRange

    load_dotenv()
    console = Console()

    # Print all available options
    print("=== CitationMode Options ===")
    for attr in dir(CitationMode):
    if not attr.startswith('_'):
    print(f"CitationMode.{attr}: {getattr(CitationMode, attr)}")

    print("\n=== ModelType Options ===")
    for attr in dir(ModelType):
    if not attr.startswith('_'):
    value = getattr(ModelType, attr)
    print(f"ModelType.{attr}: {value}")

    # If it's a class/category, explore its sub-options
    if hasattr(value, '__dict__'):
    for sub_attr in dir(value):
    if not sub_attr.startswith('_'):
    print(f" ModelType.{attr}.{sub_attr}: {getattr(value, sub_attr)}")

    print("\n=== SearchFocus Options ===")
    for attr in dir(SearchFocus):
    if not attr.startswith('_'):
    print(f"SearchFocus.{attr}: {getattr(SearchFocus, attr)}")

    print("\n=== SourceFocus Options ===")
    for attr in dir(SourceFocus):
    if not attr.startswith('_'):
    print(f"SourceFocus.{attr}: {getattr(SourceFocus, attr)}")

    print("\n=== TimeRange Options ===")
    for attr in dir(TimeRange):
    if not attr.startswith('_'):
    print(f"TimeRange.{attr}: {getattr(TimeRange, attr)}")

    print("\n" + "="*50)
    print("Starting Perplexity query...")
    print("="*50 + "\n")

    client = Perplexity(session_token=getenv("PERPLEXITY_SESSION_TOKEN"))

    with Live(Panel("", title="Waiting for answer", border_style="white"), refresh_per_second=30, vertical_overflow="visible") as live:
    for chunk in client.ask(
    query="Explain in a simplified and easy-to-understand way what a chatbot is.",
    citation_mode=CitationMode.MARKDOWN,
    model=ModelType.Pro.Gemini25Pro0605,
    save_to_library=False,
    search_focus=SearchFocus.WEB,
    source_focus=SourceFocus.WEB,
    time_range=TimeRange.ALL,
    language="en-US",
    timezone="America/New_York",
    coordinates=(-0.123, -4.567),
    ).stream():
    if chunk.last_chunk:
    current_answer = chunk.answer or ""
    live.update(Panel(current_answer, title="Receiving tokens", border_style="blue"))

    final_answer = chunk.answer or "No answer received"
    live.update(Panel(final_answer, title="Final answer", border_style="green"))