推特GPT

GPTSummarizeTweets.py

import os

import feedparser,markdown
import pandas as pd
import time
from datetime import datetime,timedelta
from dotenv import load_dotenv
from revChatGPT.V1 import Chatbot as ChatGPT

PROXY='http://127.0.0.1:7890'
load_dotenv(dotenv_path= '.env')

def getRss(user:str,hours:int=24):
  url='https://rsshub.app/twitter/user/'+user
  feeds=feedparser.parse(url)
  df = pd.DataFrame(columns=['summary', 'published'])
  for result in feeds['entries']:
    if 'summary' not in result.keys() or result['summary'].startswith('Re @'):
      continue
    summary = result['summary']
    published = datetime.fromtimestamp(time.mktime(result['published_parsed']))
    df = df._append({'summary': summary, 'published': published},ignore_index=True)
  df = df[(df['published'] >= datetime.now() - timedelta(hours=hours))]
  return df

if __name__ == '__main__':
  tweetUser='AiBreakfast'
  feeds='\n'.join(getRss(tweetUser,48)['summary'])
  print(len(feeds),feeds)
  prompt="'''%s'''"%feeds+"organize the  to a markdown post in Chinese"
  chatgptBot = ChatGPT(config={"access_token": os.environ['CHATGPT'], 'proxy': PROXY})
  reply_text=None
  for data in chatgptBot.ask(prompt):
    convId = data['conversation_id']
    reply_text = data["message"]
  try:
    chatgptBot.delete_conversation(convId)
  except:
    pass
  print(reply_text)
  html = markdown.markdown(reply_text)

  # 将 HTML 写入文件
  with open(tweetUser+".html", "w", encoding="utf-8") as f:
    f.write(html)