update the ui and the main and req

main.py

@@ -20,7 +20,6 @@ import fitz
 
 from huggingface_hub import snapshot_download
 from transformers import (
-    TFAutoModelForVision2Seq, AutoProcessor,
     AutoTokenizer, AutoModelForSeq2SeqLM,
     AutoModelForCausalLM,pipeline
 )
@@ -73,21 +72,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 app.mount("/static",StaticFiles(directory='static'),'static')
 templates = Jinja2Templates(directory='templates')
 
@@ -108,55 +92,62 @@ def index(req:Request):
 
 
     
-@app.post("/caption")
-def caption(file:UploadFile=File(...)):
-    extension = file.filename.split(".")[-1]
+@app.post("/interpret")
+def interpret(file_img:UploadFile=File(...)):
+    extension = file_img.filename.split(".")[-1]
     Supported_extensions = ["png","jpg","jpeg"]
     if extension not in Supported_extensions:
-        return {"error": "Unsupported file type"}
-    image = Image.open(file.file)
+        return JSONResponse(content={"error": "Unsupported file type"},status_code=400)
+    image = Image.open(file_img.file)
     global interpreter
+    try:
+        caption = interpreter(image)
+    except Exception as exp:
+        return JSONResponse(content={"error": "Can't interpret the image "},status_code=400)
     
-    caption = interpreter(image)
-    
-    return {"caption": caption[0]['generated_text']}   
+    return  JSONResponse(content={"caption": caption[0]['generated_text']},status_code=200)
 
 @app.post("/summerize")
 def summerzation(file:UploadFile=File(...)):
-    
-    extension = file.filename.split(".")[-1]
-    if extension == "pdf":
-        text = get_text_from_PDF(file.file)
-    elif extension == "docx":
-        text = get_text_from_DOC(file.file)
-    elif extension == "pptx":
-        text = get_text_from_PPT(file.file)
-    elif extension == "xlsx":
-        text = get_text_from_EXCEL(file.file)
-    else:
-        return {"error": "Unsupported file type"}
-    
-    print(f'[INFO] text_extracted : {text}' )
-    
-    result=""
-    global summarizer
-    for i in range(0, len(text), 1024):
-        try:
-            summary = summarizer(text[i:i+1024], max_length=150, min_length=30, do_sample=False)
-            result += summary[0]['summary_text']
-        except Exception as e:
-            return {"error": f"Summarization failed: {str(e)}"}
-    return {"summary": result}
+    try:
+        extension = file.filename.split(".")[-1]
+        if extension == "pdf":
+            text = get_text_from_PDF(file.file)
+        elif extension == "docx":
+            text = get_text_from_DOC(file.file)
+        elif extension == "pptx":
+            text = get_text_from_PPT(file.file)
+        elif extension == "xlsx":
+            text = get_text_from_EXCEL(file.file)
+        else:
+            return JSONResponse(content={"error": "Unsupported file type"},status_code=400)
+        
+        if not text.strip():
+            return JSONResponse(content={'error':'File is emplty'},status_code=400)
+        
+        result=""
+        global summarizer
+        for i in range(0, len(text), 1024):
+            try:
+                summary = summarizer(text[i:i+1024], max_length=150, min_length=30, do_sample=False)
+                result += summary[0]['summary_text']
+            except Exception as e:
+                return JSONResponse(content={"error": f"Summarization failed: {str(e)}"},status_code=403)
+        return JSONResponse(content={"summary": result},status_code=200)
+    except Exception as exp:
+        return JSONResponse(content={"error":"Internel Server Error:"+str(exp)} ,status_code=500) 
 
 
 @app.post("/plot")
-def plot(prompt:str,file:UploadFile=File(...)):
+def plot(user_need:str,file:UploadFile=File(...)):
     try:
         extension = file.filename.split(".")[-1]
         Supported_extensions = ["xlsx","xls"]
         if extension not in Supported_extensions:
-            return {"error": "Unsupported file type"}
-        df = pd.read_excel(file.file)
+            return JSONResponse(content={"error": "Unsupported file type"},status_code=400)
+        
+        df = pd.read_excel(io=file.file)
+        
         message = f"""
 You are a helpful assistant that helps users write Python code.
 ## Requirements:
@@ -173,6 +164,7 @@ df.dtypes:{df.dtypes.to_dict()}
 -don't use any other libraries except pandas, matplotlib, seaborn.
 -don't use any other functions except the ones provided in the libraries.
 -don't write the code for the dataframe creation.
+-check if the columns has a nan values and raise exception if yes .
 -exclude plt.show() from the code.
 -you have to write the code in a markdown code block.
 -make sure that the type of the chart is compatible with the dtypes of the columns 
@@ -181,14 +173,14 @@ df.dtypes:{df.dtypes.to_dict()}
 -if the user task is not clear or there is an error like the column names are not in the dataframe, raise an 
 error.
 
-##Prompt: {prompt}.
+##Prompt: {user_need}.
         """
         global generator
         output = generator(message, max_length=1000)
         match = re.search(r'```python(.*?)```', output[0]["generated_text"], re.DOTALL)
         code =''
         if not match:
-          return {"error": "Can't generate the plot"}
+          return JSONResponse(content={"error": "Can't generate the plot"},status_code=403)
           
         code = match.group(1).replace("plt.show()\n","")
         
@@ -204,19 +196,20 @@ error.
             plt.savefig(buf, format='png')
             buf.seek(0)
             base64_image = base64.b64encode(buf.getvalue()).decode('utf-8')
-            return {"plot": f"data:image/png;base64,{base64_image}"}
+            return JSONResponse(content={"plot": f"data:image/png;base64,{base64_image}"},status_code=200)
         except Exception as e:
-            return {"error": str(e)}
+            print(e)
+            return JSONResponse(content={"error": str(e) },status_code=500)
     except Exception as exp:
-        return {"error":"Internel Server Error:"+str(exp)}        
+        return JSONResponse(content={"error":"Internel Server Error:"+str(exp)} ,status_code=500)       
 
 
 
 
 
 
-def get_text_from_PDF(file):
-  doc = fitz.open(file, filetype="pdf")
+def get_text_from_PDF(file_content):
+  doc = fitz.open(stream=file_content, filetype="pdf")
   text = ""
   for page in doc:
       text += page.get_text()
@@ -240,6 +233,6 @@ def get_text_from_DOC(file):
   return text
 
 def get_text_from_EXCEL(file):
-  df = pd.read_excel(file)
+  df = pd.read_excel(io=file)
   text = df.to_string()
   return text

requirements.txt

@@ -4,6 +4,7 @@ python-docx
 python-pptx
 pandas
 python-multipart
+openpyxl
 PyMuPDF
 matplotlib
 transformers

static/scripts/dataViz.js

@@ -72,12 +72,12 @@ label.addEventListener("drop", (e) => {
 });
 form.addEventListener("submit", (e) => {
 	e.preventDefault();
-	const prompt = prompt_txt.value.trim();
+	/*const prompt = prompt_txt.value.trim();
 	if (prompt && file) {
 		showchart("");
 	} else {
 		console.log("no file selected or no prompted");
-	}
+	}*/
 });
 const showchart = (url) => {
 	chart_wrapper.innerHTML = `

static/scripts/image_iterpretation.js

@@ -75,17 +75,21 @@ label.addEventListener("drop", (e) => {
 });
 form.addEventListener("submit", async (e) => {
 	e.preventDefault();
-	let res = await new Promise((resolve, reject) => {
-		setTimeout(() => {
-			resolve({
-				src: "../static/assests/1694502952506.jpg",
-				discription: "this is a discription",
-			});
-		}, 3000);
-	});
-	const img = document.createElement("img");
-	img.src = res.src;
-	img_wrapper.appendChild(img);
-	text_result.innerText = res.discription;
-	result_img.scrollIntoView({ behavior: "smooth" });
+	if (!file) {
+		toast("No Image selected ...", "warning");
+	} else {
+		const data = new FormData();
+		data.append("file_img", file);
+		const response = await fetch("/interpret", {
+			method: "post",
+			body: data,
+		});
+		const result = await response.json();
+		if (result.error) {
+			toast(result.error, "error");
+			console.log(result.error);
+			return;
+		}
+		text_result.innerHTML = result.caption;
+	}
 });

static/scripts/summerize.js

@@ -63,11 +63,10 @@ form.addEventListener("submit", async (e) => {
 	/*if (file === null) {
     console.error("No file choosed");
   } else {
-    const formData = new FormData();
+    /*const formData = new FormData();
     formData.append("file", file);
     const res = await fetch("/summarize", {
       method: "post",
-
       body: formData,
     });
     const data = await res.json();
@@ -99,7 +98,6 @@ const typing = (text, element, parent) => {
 		if (textsplit.length > 0) {
 			element.innerHTML += `${textsplit.shift()} `;
 			parent.scrollIntoView({ behavior: "smooth" });
-
 			element.scrollIntoView({ behavior: "smooth" });
 		} else {
 			clearInterval(interval);

static/styles/notification.css

@@ -12,7 +12,7 @@
 
 .notification {
     position: relative;
-    width: 350px;
+    width: 300px;
     display: flex;
     align-items: center;
     gap: 20px;