Update app.py

app.py

@@ -73,7 +73,7 @@ def get_accounts() -> List[dict]:
 
 
 @app.get("/tweets/{username}")
-def get_tweets(username: str) -> dict:
+async 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)
@@ -81,13 +81,17 @@ def get_tweets(username: str) -> dict:
     #     return {"detail": "Account not in scope of project."}
     if username in username_list:
         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)
+        if isinstance(df_tweets, pd.DataFrame):
+            print(df_tweets.head(2))
+            print(df_tweets.shape)
+            df_tweets = df_tweets[['handle','created_at','full_text']]
+            df_tweets = df_tweets.sort_values('created_at',ascending=True).tail(10)
+            df_tweets_html = df_tweets.to_html(classes="center", index=False)
+
+            return HTMLResponse(content=df_tweets_html, status_code=200)
+        else:
+            print("Error: Failed to retrieve tweets.")
+            return df_tweets
     else:
         return {"detail": "Account not in scope of project."}