chore: update

.gitignore

@@ -0,0 +1,327 @@
+# Created by https://www.toptal.com/developers/gitignore/api/jupyternotebooks,python,pycharm+all,visualstudiocode,macos
+# Edit at https://www.toptal.com/developers/gitignore?templates=jupyternotebooks,python,pycharm+all,visualstudiocode,macos
+
+### JupyterNotebooks ###
+# gitignore template for Jupyter Notebooks
+# website: http://jupyter.org/
+
+.ipynb_checkpoints
+*/.ipynb_checkpoints/*
+
+# IPython
+profile_default/
+ipython_config.py
+
+# Remove previous ipynb_checkpoints
+#   git rm -r .ipynb_checkpoints/
+
+### macOS ###
+# General
+.DS_Store
+.AppleDouble
+.LSOverride
+
+# Icon must end with two \r
+Icon
+
+
+# Thumbnails
+._*
+
+# Files that might appear in the root of a volume
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+
+# Directories potentially created on remote AFP share
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+
+### macOS Patch ###
+# iCloud generated files
+*.icloud
+
+### PyCharm+all ###
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm and Rider
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# AWS User-specific
+.idea/**/aws.xml
+
+# Generated files
+.idea/**/contentModel.xml
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/artifacts
+# .idea/compiler.xml
+# .idea/jarRepositories.xml
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+# *.iml
+# *.ipr
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# SonarLint plugin
+.idea/sonarlint/
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+
+# Android studio 3.1+ serialized cache file
+.idea/caches/build_file_checksums.ser
+
+### PyCharm+all Patch ###
+# Ignore everything but code style settings and run configurations
+# that are supposed to be shared within teams.
+
+.idea/*
+
+!.idea/codeStyles
+!.idea/runConfigurations
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+
+# IPython
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+### Python Patch ###
+# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration
+poetry.toml
+
+# ruff
+.ruff_cache/
+
+# LSP config files
+pyrightconfig.json
+
+### VisualStudioCode ###
+.vscode/*
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+!.vscode/*.code-snippets
+
+# Local History for Visual Studio Code
+.history/
+
+# Built Visual Studio Code Extensions
+*.vsix
+
+### VisualStudioCode Patch ###
+# Ignore all local history of files
+.history
+.ionide
+
+# End of https://www.toptal.com/developers/gitignore/api/jupyternotebooks,python,pycharm+all,visualstudiocode,macos

main.py

@@ -0,0 +1,98 @@
+import logging
+import os
+import sys
+import threading
+
+sys.path.insert(0, "src")
+os.environ['TRANSFORMERS_CACHE'] = '/tmp'
+
+from telegram.ext import (
+    CommandHandler,
+    CallbackContext,
+    Application,
+    ContextTypes,
+)
+from api import GirlfriendGPT
+from functools import partial  
+from typing import List
+from telegram import Update
+from termcolor import colored
+
+SELECT_COMMAND, GET_TEXT = range(2)
+
+# Enable logging
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+# def show_results(response_messages):
+#     print(colored("\nResults: ", "blue", attrs=["bold"]))
+#     for message in response_messages:
+#         if message.mime_type:
+#             print(message.url, end="\n\n")
+#         else:
+#             print(message.text, end="\n\n")
+
+
+def shutdown(updater: Update):
+    updater.stop()
+    updater.is_idle = False
+
+def stop(update):
+    threading.Thread(target=shutdown(updater=update)).start()
+
+async def hello(update: Update, context: CallbackContext) -> None:
+    intro_text = f"🤖 Greetings human!🤗\nI'm a bot built by Rexthecoder\n🦾 I can do a lot of things"
+    await update.message.reply_text(intro_text)
+
+
+# async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
+#     """Echo the user message."""
+#     await update.message.reply_text("Enter your text")
+#     return GET_TEXT
+
+
+class LoggingDisabled:
+    """Context manager that turns off logging within context."""
+
+    def __enter__(self):
+        logging.disable(logging.CRITICAL)
+
+    def __exit__(self, exit_type, exit_value, exit_traceback):
+        logging.disable(logging.NOTSET)
+
+
+def main():
+    app = Application.builder().token(
+        '6207542226:AAEoNVXw766dq8fYqVvBQW9Hve2Rovq0ERI',).build()
+    # application = Application.builder().token('6207542226:AAGXBVVxNNUKLUz17-5_sGJnhFDaVWUXTc8').build()
+    # application.add_handler(CommandHandler('start', hello))
+
+    # # Run the bot until the user presses Ctrl-C
+    # application.run_polling()
+    run_agent(
+        agent=GirlfriendGPT(
+            token="hello",
+            application=app
+        )
+    )
+    print(f"Starting Agent...")
+
+
+def run_agent(agent: GirlfriendGPT, as_api: bool = False) -> None:
+    # For Debugging
+    summary_handler = agent.conversation_summary_handler()
+    agent.application.add_handler(summary_handler)
+    agent.application.add_handler(CommandHandler('stop', stop))
+    agent.application.add_handler(CommandHandler('start', hello))
+    # agent.application.add_handler(CommandHandler('summary', agent.conversation_summary))
+    # agent.application.add_handler(MessageHandler(
+    #     filters.TEXT & ~filters.COMMAND, agent.create_response))
+    agent.application.run_polling()
+    # agent.application.add_handler(MessageHandler(
+    #     filters.TEXT & ~filters.COMMAND, agent.create_response))
+
+
+if __name__ == "__main__":
+    main()

src/agent/base.py

@@ -0,0 +1,46 @@
+"""Define your LangChain chatbot."""
+import re
+from abc import abstractmethod
+
+from telegram import Update
+import telegram
+from telegram.ext import (
+    ConversationHandler,
+    CallbackContext,
+)
+
+from agent.tools.text_summary import ConversationSummary
+
+SELECT_COMMAND, GET_TEXT = range(2)
+
+
+class LangChainAgentBot(ConversationSummary):
+    def is_verbose_logging_enabled(self):
+        return True
+
+    def send_message(self, message: str, update: Update) -> str:
+        """Send a message to Telegram.
+
+        Note: This is a private endpoint that requires authentication."""
+        update.message.reply_text(message)
+        return "ok"
+
+    def _invoke_later(self, delay_ms: int, message: str, chat_id: str):
+        self.invoke_later(
+            "send_message",
+            delay_ms=delay_ms,
+            arguments={
+                "message": message,
+                "chat_id": chat_id,
+            },
+        )
+
+    async def cancel(update: Update, context: CallbackContext) -> int:
+        """Cancel the conversation."""
+        await update.message.reply_text("Oops, glad to help you.")
+        return ConversationHandler.END
+
+
+    async def create_response(self, update: Update, context: CallbackContext) -> None:
+        await update.message.chat.send_action(action=telegram.constants.ChatAction.TYPING)
+        await update.message.reply_text(update.message.text)

src/agent/tools/text_summary.py

@@ -0,0 +1,65 @@
+
+import logging
+import telegram
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from telegram.ext import (
+    Updater,
+    CommandHandler,
+    ConversationHandler,
+    CallbackContext,
+    CallbackQueryHandler,
+    MessageHandler,
+    filters
+)
+
+from telegram import Update
+
+NAME = "ConversationSummary"
+
+DESCRIPTION = """
+Useful for summarizing conversation. 
+Input: A conversation text
+Output: A summazrised version of the conversation
+"""
+
+SELECT_COMMAND, GET_TEXT = range(2)
+
+
+class ConversationSummary():
+    """Tool used to summarize text from a conversational text."""
+
+    tokenizer = AutoTokenizer.from_pretrained(
+        "mrm8488/flan-t5-small-finetuned-samsum")
+
+    model = AutoModelForSeq2SeqLM.from_pretrained(
+        "mrm8488/flan-t5-small-finetuned-samsum")
+
+    async def summarize(self, input: str, words: int):
+
+        logging.info(f"{input} {words}")
+
+        input_ids = self.tokenizer(input, return_tensors="pt").input_ids
+        outputs = self.model.generate(input_ids, max_length=words)
+        decoded_output = self.tokenizer.decode(
+            outputs[0], skip_special_tokens=True)
+        return f"{decoded_output}"
+
+    def conversation_summary_handler(self) -> ConversationHandler:
+        handler = ConversationHandler(
+            entry_points=[CommandHandler(
+                "summary", self.conversation_summary)],
+            states={
+                GET_TEXT: [MessageHandler(filters.TEXT & ~filters.COMMAND, self.process_conversation_summary)],
+            },
+            fallbacks=[CommandHandler("cancel", self.cancel)],
+        )
+        return handler
+
+    async def process_conversation_summary(self, update: Update, context: CallbackContext) -> int:
+        message = update.message.text
+        await self.summarize(message, 100)
+        return ConversationHandler.END
+
+    async def conversation_summary(self, update: Update, context: CallbackContext) -> str:
+        await update.message.reply_text(f'Please enter your conversations...')
+        return GET_TEXT

src/api.py

@@ -0,0 +1,63 @@
+"""Scaffolding to host your LangChain Chatbot on Steamship and connect it to Telegram."""
+from typing import List, Optional, Type
+
+from pydantic import Field
+
+
+from agent.base import LangChainAgentBot 
+from telegram.ext import Updater, CommandHandler, CallbackContext, Application, ContextTypes
+from telegram import Update
+
+
+VERBOSE = True
+
+
+class GirlFriendAIConfig():
+    elevenlabs_api_key: str = Field(
+        default="", description="Optional API KEY for ElevenLabs Voice Bot"
+    )
+    elevenlabs_voice_id: str = Field(
+        default="", description="Optional voice_id for ElevenLabs Voice Bot"
+    )
+
+
+class GirlfriendGPT(LangChainAgentBot):
+    """Deploy LangChain chatbots and connect them to Telegram."""
+
+    token: str
+    application: Application
+
+    def __init__(self, token, application):
+        super().__init__()
+        self.application = application
+        # application.add_handler(CommandHandler('start', hello))
+        # Run the bot until the user presses Ctrl-C
+        # self.application.run_polling()
+        self.token = token
+
+    # async def echo(self, update: Update, context: CallbackContext) -> None:
+    #     """Echo the user message."""
+    #     await update.message.reply_text(update.message.text)
+
+    # def voice_tool(self) -> Optional[Tool]:
+    #     """Return tool to generate spoken version of output text."""
+    #     # return None
+    #     return GenerateSpeechTool(
+    #         client=self.client,
+    #         voice_id=self.config.elevenlabs_voice_id,
+    #         elevenlabs_api_key=self.config.elevenlabs_api_key,
+    #     )
+
+    # def get_memory(self, chat_id):
+    #     if self.context and self.context.invocable_instance_handle:
+    #         my_instance_handle = self.context.invocable_instance_handle
+    #     else:
+    #         my_instance_handle = "local-instance-handle"
+    #     memory = ConversationBufferMemory(
+    #         memory_key="chat_history",
+    #         chat_memory=ChatMessageHistory(
+    #             client=self.client, key=f"history-{chat_id}-{my_instance_handle}"
+    #         ),
+    #         return_messages=True,
+    #     )
+    #     return memory