Update app.py

app.py

@@ -1,62 +1,46 @@
 import gradio as gr
-import os
-import openai
+import requests
 from dotenv import load_dotenv
+from telegram import Bot
 
 # Load environment variables
 load_dotenv()
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+bot = Bot(token=TELEGRAM_BOT_TOKEN)
 
-# Initialize OpenAI client
-openai.api_key = OPENAI_API_KEY
-openai.api_base = "https://api.aimlapi.com/"
-
-# Greeting function for Gradio
-def greet(name):
-    return f"Hello {name}!"
+def send_message_to_user(chat_id, message):
+    """Send a test message to a user via the bot."""
+    try:
+        bot.send_message(chat_id=chat_id, text=message)
+        return f"Message sent to chat ID {chat_id}"
+    except Exception as e:
+        return f"Failed to send message: {str(e)}"
 
-# OpenAI Chat Completion Function
-def get_openai_response(user_content, model="o1-preview", max_tokens=500):
+def check_bot_status():
+    """Check if the bot is online."""
     try:
-        chat_completion = openai.ChatCompletion.create(
-            model=model,
-            messages=[
-                {"role": "user", "content": user_content}
-            ],
-            max_tokens=max_tokens
-        )
-        response = chat_completion.choices[0].message['content']
-        return response
+        response = bot.get_me()
+        return f"Bot is online: {response.username}"
     except Exception as e:
-        return f"An error occurred: {e}"
+        return f"Bot is offline: {str(e)}"
 
-# Gradio interface
-def openai_chat_interface(user_content, model, max_tokens):
-    return get_openai_response(user_content, model, max_tokens)
+# Define the Gradio interface
+with gr.Blocks() as dashboard:
+    gr.Markdown("# Telegram Bot Admin Dashboard")
 
-# Define Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# OpenAI Chat Completion & Greeting Interface")
-    
-    # Greeting section
+    # Bot status check
     with gr.Row():
-        gr.Markdown("### Greeting Section")
-        greet_input = gr.Textbox(label="Enter your name")
-        greet_output = gr.Textbox(label="Greeting Output")
-        greet_button = gr.Button("Greet")
-        greet_button.click(greet, inputs=greet_input, outputs=greet_output)
-    
-    # OpenAI Chat Completion Section
+        bot_status_button = gr.Button("Check Bot Status")
+        bot_status_output = gr.Textbox(label="Bot Status")
+        bot_status_button.click(check_bot_status, inputs=None, outputs=bot_status_output)
+
+    # Send a message
     with gr.Row():
-        gr.Markdown("### OpenAI Chat Completion")
-        user_input = gr.Textbox(label="Enter your message")
-        model_input = gr.Radio(["o1-preview", "o1-mini"], label="Model Selection", value="o1-preview")
-        max_tokens_input = gr.Slider(1, 7000, value=500, label="Max Tokens")
-        openai_response = gr.Textbox(label="OpenAI Response")
-        chat_button = gr.Button("Get OpenAI Response")
-        chat_button.click(openai_chat_interface, 
-                          inputs=[user_input, model_input, max_tokens_input], 
-                          outputs=openai_response)
+        chat_id_input = gr.Textbox(label="Chat ID")
+        message_input = gr.Textbox(label="Message to Send")
+        send_message_button = gr.Button("Send Message")
+        send_message_output = gr.Textbox(label="Send Message Output")
+        send_message_button.click(send_message_to_user, inputs=[chat_id_input, message_input], outputs=send_message_output)
 
-# Launch Gradio app
-demo.launch()
+# Launch the dashboard
+dashboard.launch()