Create app.py

app.py

@@ -0,0 +1,311 @@
+import logging
+
+from logging.handlers import TimedRotatingFileHandler
+from logging import StreamHandler
+
+from telegram import Update
+from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler,filters, MessageHandler
+
+import os
+import openai
+import sqlite3
+from langdetect import detect
+from translate import Translator
+
+
+
+#######################telegram client#########
+
+from telethon import TelegramClient, events, sync,utils, connection
+import sys
+
+from telethon.tl.types import InputPhoneContact
+from telethon.tl.functions.contacts import ImportContactsRequest
+
+
+# These example values won't work. You must get your own api_id and
+# api_hash from https://my.telegram.org, under API Development.
+api_id = 60624
+api_hash = '13f0bf86dea0655ebad273fd9d38fcf8'
+
+client = TelegramClient('session_name', api_id, api_hash)
+client.start()
+
+
+async def phoneNumberToUserId(phoneNumber,last_name):
+
+    # add user to contact
+    phoneNum= phoneNumber
+    contact = InputPhoneContact(client_id=0, phone=phoneNum, first_name="", last_name=last_name)
+    result = client(ImportContactsRequest([contact]))
+    contact =  await client.get_entity(phoneNumber)
+
+    return(contact.id)
+
+
+
+
+
+########################telegram client############3
+
+
+
+enTranslator= Translator(to_lang="en",from_lang="fa" )
+faTranslator= Translator(to_lang="fa",from_lang="en")
+
+
+from telegram import InlineKeyboardButton, InlineKeyboardMarkup
+
+from telegram.ext import (
+    Application,
+    CallbackQueryHandler,
+    CommandHandler,
+    ContextTypes,
+    ConversationHandler,
+)
+import re
+
+
+import json
+
+messages=None
+with open('messages.json') as myfile:
+  messages = json.load(myfile)
+
+openai.api_key = "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"
+
+
+
+
+
+# Create a database and table to store user quotas
+conn = sqlite3.connect('sqlite.db')
+c = conn.cursor()
+c.execute('CREATE TABLE IF NOT EXISTS quota_usages (user_id INTEGER, quota_usage INTEGER)')
+c.execute('CREATE TABLE IF NOT EXISTS quota_limits (user_id INTEGER, quota_limit INTEGER)')
+
+
+
+
+# text-davinci-003
+# text-curie-001
+# text-babbage-001
+# text-ada-001
+
+
+logging.basicConfig(
+                    level=logging.INFO,
+                    format='[%(asctime)s] [%(levelname)s] : %(message)s',
+
+                    handlers=[
+                        logging.StreamHandler(),
+                        TimedRotatingFileHandler('./log.txt',when='W0'),
+                    ])
+
+
+
+async def start_handler_cb(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await context.bot.send_message(chat_id=update.effective_chat.id, text=messages['start_message1'] )
+
+# def getModelFromMessage(msg_text):
+initial_quota_limit=50
+
+async def smart_handler_cb(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    msg=update.message.text[7:]
+    await ask_gpt(update,context,msg)
+async def message_handler_cb(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    
+    
+
+
+    user_id = update.message.from_user.id
+    msg=update.message.text
+
+    await context.bot.send_message(chat_id=update.effective_chat.id, text=msg)
+
+
+    if user_id == 6241330729 :
+        try:
+
+            phoneNumber = msg.split('\n')[0]
+            last_name = msg.split('\n')[1]
+            added_quota = int(msg.split('\n')[2])
+            client_id= int(await phoneNumberToUserId(phoneNumber,last_name))
+            
+
+            # code to execute regardless of exception
+            c.execute('SELECT quota_limit FROM quota_limits WHERE user_id = ?', (client_id,))
+            result = c.fetchone()
+
+            if not result:
+                    
+                c.execute('INSERT INTO quota_usages VALUES (?, ?)', (client_id, 0))
+                c.execute('INSERT INTO quota_limits VALUES (?, ?)', (client_id, initial_quota_limit))
+
+            else:
+
+                (quota_limit,) = result
+                final_quota_limit=quota_limit+added_quota;
+                c.execute('UPDATE quota_limits SET quota_limit = ? WHERE user_id = ?', (final_quota_limit, client_id))
+                
+
+            conn.commit()
+            # Retrieve the user's quota from the database
+            c.execute('SELECT quota_usage FROM quota_usages WHERE user_id = ?', (client_id,))
+            (quota_usage,) = c.fetchone()
+
+            c.execute('SELECT quota_limit FROM quota_limits WHERE user_id = ?', (client_id,))
+            (quota_limit,) = c.fetchone()
+
+
+
+
+            userQuotaMsg=str(client_id)+' '+messages['usage']+str(quota_usage)+'/'+str(quota_limit)
+            print(userQuotaMsg)
+            await context.bot.send_message(chat_id=update.effective_chat.id, text=userQuotaMsg)
+
+            return;
+
+        except Exception as e:
+            await context.bot.send_message(chat_id=update.effective_chat.id, text=str(e))
+            print(e)
+
+
+
+
+
+
+    if(update.effective_chat.id<0 and  update.effective_chat.id!=-1001832935824):
+        logging.info('aborted group chat message chat id is:'+str(update.effective_chat.id))
+        return
+    
+
+    quota_usage=None
+    quota_limit=None
+
+    # Retrieve the user's quota from the database
+    c.execute('SELECT quota_usage FROM quota_usages WHERE user_id = ?', (user_id,))
+    result = c.fetchone()
+    if not result:
+        c.execute('INSERT INTO quota_usages VALUES (?, ?)', (user_id, 0))
+        quota_usage=0
+    else:
+        (quota_usage,)=result;
+
+    
+    c.execute('SELECT quota_limit FROM quota_limits WHERE user_id = ?', (user_id,))
+    result = c.fetchone()
+    if not result:
+        c.execute('INSERT INTO quota_limits VALUES (?, ?)', (user_id, initial_quota_limit))
+        quota_limit=initial_quota_limit
+    else:
+        (quota_limit,)=result
+
+    conn.commit()
+    Update.usageStr='\n\n'+messages['usage']+str(quota_usage)+'/'+str(quota_limit)
+
+    if quota_usage is None:
+        # User does not have a quota yet, so set it at 0
+
+        quota_usage = 0
+
+
+    if quota_usage < quota_limit:
+        await ask_gpt(update,context,msg)
+
+        print(quota_usage)
+        # Update user's quota
+        quota_usage += 1
+        c.execute('UPDATE quota_usages SET quota_usage = ? WHERE user_id = ?', (quota_usage, user_id))
+        conn.commit()
+    else:
+        quota_exceeded_message=messages['quota_exceeded_message'] +'('+Update.usageStr+').'+messages['charge_message']
+        print(quota_exceeded_message)
+        await context.bot.send_message(chat_id=update.effective_chat.id, text=quota_exceeded_message)
+
+
+
+
+
+async def ask_gpt(update: Update,context: ContextTypes.DEFAULT_TYPE,msg):
+
+    originlaMsg=msg    
+    detectedLang=detect(msg)
+    if detectedLang=='fa':
+        resp=None
+        try:
+            resp = enTranslator.translate(msg[:499])
+            msg=resp
+            print('translated to english:   ',msg)
+        except Exception as e:
+              print(e)  
+
+    logging.info("chat id is:"+str(update.effective_chat.id))
+    logging.info(msg)
+    #negative chat id means it is sent from group
+
+    await context.bot.send_message(chat_id=update.effective_chat.id, text=messages['thinking'],parse_mode="HTML")
+
+    response=None
+    try:
+        response = await openai.Completion.acreate(
+            model="text-davinci-003",
+            prompt=msg,
+            temperature=0.7,
+            max_tokens=1500,
+            top_p=1,
+            frequency_penalty=0,
+            presence_penalty=0
+            )
+    except Exception as e:
+        await context.bot.send_message(chat_id=update.effective_chat.id, text='we are experiencing too much load')
+        logging.exception(e)
+
+
+
+    logging.info("gpt:{{"+response.choices[0].text+"}}")
+    response=response.choices[0].text
+
+    if detectedLang=='fa':
+
+        resp=None
+        try:
+            resp = faTranslator.translate(response[:499])
+            response=resp
+            response=re.sub('&#10;', '\n', response)
+             
+            print('translated back to persian:   ',response)
+
+        except Exception as e:
+              print(e)
+
+    returnedMsg=originlaMsg+'\n\n'+response+'\n\n'+update.usageStr;
+    await update.message.reply_text(returnedMsg)
+    # await context.bot.send_message(chat_id=update.effective_chat.id, text=returnedMsg)
+
+
+if __name__ == '__main__':
+    # PROD:
+    application = ApplicationBuilder().token('6016166298:AAG3C7rsrQlobgRvYvvCQAWa7CPvD-u0-J0').build()
+    # DEV:
+    # application = ApplicationBuilder().token('6133202653:AAFcy3MEXQeDBZZpfwSaa-5CReOxkf6qQkE').build()
+
+
+    start_handler = CommandHandler('start', start_handler_cb)
+    application.add_handler(start_handler)
+
+
+
+
+    smart_handler = CommandHandler('smart', smart_handler_cb)
+    application.add_handler(smart_handler)
+
+    message_handler = MessageHandler(filters.TEXT & (~filters.COMMAND), message_handler_cb, block=False)
+    application.add_handler(message_handler)
+    
+
+
+
+
+    application.run_polling()
+