Update main.py

main.py

@@ -2,7 +2,6 @@ from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import HTMLResponse
 import pandas as pd
 import io
-import requests
 import time
 
 app = FastAPI()
@@ -45,6 +44,15 @@ async def upload_file(file: UploadFile = File(...)):
         cum_perc=lambda df: df['perc'].cumsum(),
     )
 
+    dataframe_styled = (
+        ip_address_count_df.head(1000)
+        .style.background_gradient(subset=['count', 'perc', 'cum_perc'], cmap='cividis')
+        .format({'count': '{:,}', 'perc': '{:.1%}', 'cum_perc': '{:.1%}'})
+    )
+    
+    # Render the styled DataFrame to HTML
+    styled_html = dataframe_styled.render()
+
     result = (
         "<h1>Redundant IP Requests....</h1>"
         "<p>The Total API Requests from the sample logs are : {total_requests}</p>"
@@ -55,17 +63,7 @@ async def upload_file(file: UploadFile = File(...)):
         total_requests=logs_df.shape[0],
         redundant_requests=ip_address_count_df.shape[0],
         redundant_percentage=(ip_address_count_df.shape[0] / logs_df.shape[0]) * 100,
-       dataframe_html = (
-    ip_address_count_df.head(1000)
-    .style.background_gradient(subset=['count', 'perc', 'cum_perc'], cmap='cividis')
-    .format({'count': '{:,}', 'perc': '{:.1%}', 'cum_perc': '{:.1%}'})
-    .render()
-    .replace('\n', '')  # Remove newline characters to format the HTML properly
-)
+        dataframe_html=styled_html,
     )
 
-    # # Save result in a new HTML file
-    # with open("result.html", "w") as f:
-    #     f.write(result)
-
-    return result
+    return HTMLResponse(content=result)