Update main.py

main.py

@@ -1,56 +1,23 @@
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import HTMLResponse
-from fastapi import FastAPI
-import json
-import requests
+from fastapi import FastAPI, File, UploadFile
 import csv
-from fastapi import FastAPI, UploadFile, File
-from fastapi.responses import HTMLResponse
-from fastapi.responses import HTMLResponse, FileResponse
+from typing import List
+from io import TextIOWrapper
+
+app = FastAPI()
+
 # Dictionaries to store URLs, API counts, and associated names
 url_counts = {}
 api_counts = {}
 url_names = {}
-
-app = FastAPI()
-
-def ask(query):
-    headers = {"Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoiM2NlMTljZDItNTk2Zi00NzRhLWE3OGMtZWZhNTM1YWJlNmY0IiwidHlwZSI6ImFwaV90b2tlbiJ9.x_PPBluoOy2GT3ZjDQ3dcS8WtbNn95RfkoFKkhM_a5A"}  # Replace with your API key
-    url = "https://api.edenai.run/v2/text/chat"
-
-    # Create the content dictionary
-    content ="The inventory consists of various items, including Milk (50 units in stock, 10 units pending), Bread (100 units in stock, 20 units pending), Eggs (75 dozen in stock, 5 dozen pending), Rice (200 pounds in stock, 15 pounds pending), Chicken (40 pounds in stock, 8 pounds pending), Fresh Vegetables (various in stock, 30 pounds pending), Canned Soup (150 cans in stock, 25 cans pending), Pasta (80 pounds in stock, 10 pounds pending), Toilet Paper (60 rolls in stock, 12 rolls pending), and Shampoo (30 bottles in stock, 5 bottles pending)."
-    # Concatenate the input query with the "text" field in the payload
-    payload = {
-        "providers": "openai",
-        "text": "from the summary: ",
-        "chatbot_global_action": content + " " + query,
-        "previous_history": [],
-        "temperature": 0.0,
-        "max_tokens": 300
-    }
-
-    response = requests.post(url, json=payload, headers=headers)
-    print(response)
-    result = json.loads(response.text)
-    print(result)
-    generated_text = result['openai']['generated_text']
-    print(generated_text)
-    return generated_text
-
-@app.get("/ask")
-async def ask_question(question_text: str):
-    # Call the question function with the provided question_text
-    response = ask(question_text)
-    return {"response": response}
+api_names = {}
 
 @app.post("/upload/")
-async def upload_file(file: UploadFile = File(...)):
-    contents = await file.read()
-    with open(contents, 'r', newline='', encoding='utf-8') as csvfile:
+async def upload_csv(file: UploadFile):
+    if file.content_type == "text/csv":
+        total_records = 0
+        csvfile = TextIOWrapper(file.file, encoding='utf-8')
         csvreader = csv.reader(csvfile)
         header = next(csvreader)  # Skip the header row
-        total_records = 0
         for row in csvreader:
             url = row[3]  # Assuming the URL is in the 4th column (index 3)
             api = row[4]  # Assuming the API is in the 5th column (index 4)
@@ -76,31 +43,20 @@ async def upload_file(file: UploadFile = File(...)):
         redundant_urls = [url for url, count in url_counts.items() if count > 1]
         redundant_apis = [api for api, count in api_counts.items() if count > 1]
 
-    # Print the results including names
-    print("Total GET Records:", total_records)
-
-    print("\nRedundant GET Requests:")
-    for url in redundant_urls:
-        print("URL:", url)
-        print("Name:", url_names[url])
-        print("Count:", url_counts[url])
-        print("---")
-
-    # Calculate the percentage of redundant URLs and APIs
-    percentage_redundant_urls = (len(redundant_urls) / total_records) * 100 if total_records > 0 else 0
-
-    print("\nPercentage of Redundant GET Requests: {:.2f}%".format(percentage_redundant_urls))
-    
-    
-
-
-
-    return HTMLResponse(content=percentage_redundant_urls)
-
-
-
-
-
-
-
-
+        # Calculate the percentage of redundant URLs and APIs
+        percentage_redundant_urls = (len(redundant_urls) / total_records) * 100 if total_records > 0 else 0
+
+        return {
+            "Total GET Records": total_records,
+            "Redundant GET Requests": [
+                {
+                    "URL": url,
+                    "Name": url_names[url],
+                    "Count": url_counts[url]
+                }
+                for url in redundant_urls
+            ],
+            "Percentage of Redundant GET Requests": round(percentage_redundant_urls, 2)
+        }
+    else:
+        return {"error": "Invalid file format. Please upload a CSV file."}