Sydney-Pi-Images/chatsydney_full.py at main · Manamama/Sydney-Pi-Images · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#Full UI for chatting with SydneyClient, in Markdown, with images, context and more
from rich.console import Console
from rich.markdown import Markdown

def print_markdown(md_text):
    console = Console()

    console.print(Markdown(md_text))


import asyncio

from sydney import SydneyClient

import tkinter as tk
from tkinter import filedialog

def pick_file(folder):
    root = tk.Tk()
    root.withdraw()  # Hide the main window
    file_path = filedialog.askopenfilename(
        initialdir=folder,
        #remove the above, if needed
        title="Select a local graphics file, as no URI provided as argument in the !image prompt",
        filetypes=(("Image files", "*.png *.jpg *.jpeg"), ("All files", "*.*"), ("Text and like files", "*.txt *.json *.puml"))
    )
    return file_path

async def main() -> None:
    async with SydneyClient(style="creative") as sydney:
    #or "balanced" or "precise"
        while True:
            prompt = input("User: ")

            if prompt == "!reset":
                await sydney.reset_conversation(style="creative")

                continue
            elif prompt == "!exit":
                break


            # Check if the prompt starts with "!image"
            if prompt.startswith("!image"):
                # Split the prompt into three parts: the "!image" keyword, the URI, the prompt
                _, uri, prompt = prompt.split(maxsplit=2)

                # Check if the URI starts with "http"
                if not uri.startswith("http"):
                    # If not, assume it's a local file and show the file picker
                    uri = pick_file("Pictures")
                    print ("\033[2mSelected:\033[1m", uri, "\033[5m Wait for image upload...\033[0m \n")


                print("Sydney: ", end="", flush=True)
                # Pass the URI and the prompt as the parameters
                response=await sydney.ask(prompt, attachment=uri, citations=True)
                print_markdown(response)
                #Or:
                '''
                async for response in sydney.ask_stream(prompt, attachment=uri):

                    print(response, end="", flush=True)
                    '''

            if prompt.startswith("!context"):
                # Split the prompt into three parts: the "!context" keyword, the URI, the prompt
                _, uri, prompt = prompt.split(maxsplit=2)

                # Check if the URI starts with "http"
                if not uri.startswith("http"):
                    # If not, assume it's a local file and show the file picker
                    uri = pick_file("Documents")
                    print ("\033[2mSelected:\033[1m", uri, "\033[5m Wait for context upload...\033[0m \n")

                #print ("\033[2mDone. Context to be used:\033[1m", uri, "...\033[0m \n")

                print("Sydney: ", end="", flush=True)
                # Pass the URI and the prompt as the parameters
                response=await sydney.ask(prompt, context=uri, citations=True)
                print_markdown(response)
                #Or:
                '''
                async for response in sydney.ask_stream(prompt, attachment=uri):

                    print(response, end="", flush=True)
                    '''


            else:
                # Use the ask_stream method as before
                print("Sydney: ", end="", flush=True)

                response=await sydney.ask(prompt, citations=True)
                print_markdown(response)
                 #Or:
                '''
                async for response in sydney.ask_stream(prompt):
                    print(response, end="", flush=True)
                    '''


            print("\n")


if __name__ == "__main__":
    asyncio.run(main())