diff --git a/ollachat/cli.py b/ollachat/cli.py index b0370a7..db7791b 100644 --- a/ollachat/cli.py +++ b/ollachat/cli.py @@ -7,4 +7,8 @@ def run_streamlit(): chatbot_path = os.path.join(dir_path, 'chatbot.py') # Run the command - subprocess.run(["streamlit", "run", chatbot_path]) + try: + subprocess.run(["streamlit", "run", chatbot_path]) + except KeyboardInterrupt: + print("Shutting down the Streamlit app...") + exit(0) \ No newline at end of file diff --git a/setup.py b/setup.py index 923a68d..2223fba 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ setup( name="st_ollama", - version="0.1", + version="0.1.1", author="Aziz Alto", author_email="iamaziz.alto@gmail.com", description="A Streamlit chatbot app integrating Ollama LLMs",