Initial release

main.py

@@ -0,0 +1,55 @@
+import os
+from dotenv import load_dotenv
+from flowise import Flowise, PredictionData
+from telegram import Update
+from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters
+
+global flowiseClient
+global chatflowId
+
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """Send a message when the command /start is issued."""
+    user = update.effective_user    
+    await update.message.reply_markdown_v2(
+        fr"Hi {user.mention_markdown_v2()}\!",
+    )
+
+
+async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """Handle a message."""
+    response = flowiseClient.create_prediction(PredictionData(question=update.message.text, chatflowId=chatflowId, streaming=False))
+    # Extract text from the response
+    response_text = ""
+    for item in response:
+        print(item)
+        response_text += item["text"]
+    await update.message.reply_text(response_text)
+
+if __name__ == "__main__":
+
+    # Load environment variables
+    load_dotenv()
+
+    # Setup the Flowise client
+    FLOWISE_API_KEY = os.getenv("FLOWISE_API_KEY")
+    FLOWISE_API_URL = os.getenv("FLOWISE_API_URL")
+    chatflowId = os.getenv("FLOWISE_CHATFLOW_ID")
+    if not FLOWISE_API_KEY:
+        raise ValueError("FLOWISE_API_KEY not found in environment variables")
+    if not FLOWISE_API_URL:
+        raise ValueError("FLOWISE_API_URL not found in environment variables")
+    if not chatflowId:
+        raise ValueError("FLOWISE_CHATFLOW_ID not found in environment variables")
+    
+    flowiseClient = Flowise(base_url=FLOWISE_API_URL, api_key=FLOWISE_API_KEY)
+
+    # Setup the Telegram bot
+    TELEGRAM_API_KEY = os.getenv("TELEGRAM_API_KEY")
+    if not TELEGRAM_API_KEY:
+        raise ValueError("TELEGRAM_API_KEY not found in environment variables")
+    application = Application.builder().token(TELEGRAM_API_KEY).build()
+    application.add_handler(MessageHandler(filters.TEXT, handle_message))
+    application.add_handler(CommandHandler("start", start))
+
+    # Run the bot
+    application.run_polling()