chabane commited on
Commit
6310b8b
·
1 Parent(s): fc61dfc

change in main

Browse files
Files changed (1) hide show
  1. main.py +11 -7
main.py CHANGED
@@ -1,4 +1,4 @@
1
- from fastapi import FastAPI,Request,File,UploadFile
2
  from fastapi.templating import Jinja2Templates
3
  from fastapi.staticfiles import StaticFiles
4
  from fastapi.responses import HTMLResponse,JSONResponse
@@ -110,15 +110,18 @@ def interpret(file_img:UploadFile=File(...)):
110
  @app.post("/summerize")
111
  def summerzation(file:UploadFile=File(...)):
112
  try:
113
- print(file.size)
114
  extension = file.filename.split(".")[-1]
115
- supported_ext=["pdf","xlsx","docx","ppt"]
116
  if extension not in supported_ext :
117
  return JSONResponse(content={"error": "Unsupported file type"},status_code=400)
 
 
 
 
118
  file_bytes = file.file.read()
119
 
120
- if len(file_bytes) > MAX_SIZE :
121
- return JSONResponse(content={"error": "too large file "},status_code=400)
122
  if extension == "pdf":
123
  text = get_text_from_PDF(file_bytes)
124
  elif extension == "docx":
@@ -146,7 +149,7 @@ def summerzation(file:UploadFile=File(...)):
146
 
147
 
148
  @app.post("/plot")
149
- async def plot(user_need:str,file:UploadFile=File(...)):
150
  try:
151
  extension = file.filename.split(".")[-1]
152
  Supported_extensions = ["xlsx","xls"]
@@ -158,9 +161,10 @@ async def plot(user_need:str,file:UploadFile=File(...)):
158
  if len(file_bytes) > MAX_SIZE :
159
  return JSONResponse(content={"error": "too large file "},status_code=400)
160
 
 
161
 
162
  df = pd.read_excel(io=io.BytesIO(file_bytes))
163
- print(df.head())
164
  message = f"""
165
  You are a helpful assistant that helps users write Python code.
166
  ## Requirements:
 
1
+ from fastapi import FastAPI,Request,File,UploadFile,Form
2
  from fastapi.templating import Jinja2Templates
3
  from fastapi.staticfiles import StaticFiles
4
  from fastapi.responses import HTMLResponse,JSONResponse
 
110
  @app.post("/summerize")
111
  def summerzation(file:UploadFile=File(...)):
112
  try:
113
+
114
  extension = file.filename.split(".")[-1]
115
+ supported_ext=["pdf","xlsx","docx","ppt","xls"]
116
  if extension not in supported_ext :
117
  return JSONResponse(content={"error": "Unsupported file type"},status_code=400)
118
+ if file.size > MAX_SIZE:
119
+ return JSONResponse(content={"error": "file is too large "},status_code=400)
120
+
121
+
122
  file_bytes = file.file.read()
123
 
124
+
 
125
  if extension == "pdf":
126
  text = get_text_from_PDF(file_bytes)
127
  elif extension == "docx":
 
149
 
150
 
151
  @app.post("/plot")
152
+ async def plot(user_need:str=Form(...),file:UploadFile=File(...)):
153
  try:
154
  extension = file.filename.split(".")[-1]
155
  Supported_extensions = ["xlsx","xls"]
 
161
  if len(file_bytes) > MAX_SIZE :
162
  return JSONResponse(content={"error": "too large file "},status_code=400)
163
 
164
+
165
 
166
  df = pd.read_excel(io=io.BytesIO(file_bytes))
167
+
168
  message = f"""
169
  You are a helpful assistant that helps users write Python code.
170
  ## Requirements: