Update app.py

app.py

@@ -19,6 +19,7 @@ from fastapi.templating import Jinja2Templates
 
 import scripts.sentiment as sentiment
 import scripts.twitter_scraper as ts
+from scripts.twitter_scraper import get_latest_account_tweets
 import scripts.utils as utils
 from scripts import generative
 import nltk
@@ -66,14 +67,27 @@ def get_accounts() -> List[dict]:
     df_account = df_account.format(
         {"follower_count": "{:,.0f}", "friends_count": "{:,.0f}"}
     )
-    return HTMLResponse(content=df_account.to_html(classes="center"), status_code=200)
+    html_table = df_account.to_html(classes="center", index=False)
 
+    return HTMLResponse(content=html_table, status_code=200)
 
-@app.get("/tweets/{username}", response_model=dict)
+
+@app.get("/tweets/{username}")
 def get_tweets(username: str) -> dict:
+    # if username in username_list:
+    #     query = f"from:{username} since:{start_date} until:{end_date}"
+    #     return ts.get_tweets(query=query)
+    # else:
+    #     return {"detail": "Account not in scope of project."}
     if username in username_list:
-        # query = f"from:{username} since:{start_date} until:{end_date}"
-        return ts.get_tweets(query=query)
+        df_tweets = get_latest_account_tweets(username)
+        print(df_tweets.head(2))
+        print(df_tweets.shape)
+        df_tweets = df_tweets[['handle','created_at','full_text','id']]
+        df_tweets = df_tweets.sort_values('created_at',ascending=True).tail(20)
+        df_tweets_html = df_tweets.to_html(classes="center", index=False)
+
+        return HTMLResponse(content=df_tweets_html, status_code=200)
     else:
         return {"detail": "Account not in scope of project."}