Merge branch 'main' of https://huggingface.co/spaces/MsChabane/SmartDoc

main.py

@@ -51,7 +51,7 @@ def caption(file:UploadFile=File(...)):
     if extension not in Supported_extensions:
         return {"error": "Unsupported file type"}
     image = Image.open(file.file)
-    interpreter = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
+    interpreter = pipeline("image-to-text", model="Salesforce/blip-image-captioning-large")
     caption = interpreter(image)
     return {"caption": caption[0]['generated_text']}   
 
@@ -86,30 +86,30 @@ def plot(prompt:str,file:UploadFile=File(...)):
             return {"error": "Unsupported file type"}
         df = pd.read_excel(file.file)
         message = f"""
-    You are a helpful assistant that helps users write Python code.
-    ## Requirements:
-    -you will be given a task and you will write the code to solve the task.
-    -you have a dataset called **df** contains the following information:
-    df.columns:{df.columns.to_list()}
-    df.dtypes:{df.dtypes.to_dict()}
-
-    -you have to write the code to solve the task using the dataset df.
-    -you can use pandas to manipulate the dataframe.
-    -you can use matplotlib to plot the data.
-    -you can use seaborn to plot the data.
-    -don't use print or input statements in the code.
-    -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.
-    -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 
-    -use only the column specified in the task.
-    -you have to extract the column names and the plot type  from the prompt bellow  and use them in the code.
-    -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}.
+You are a helpful assistant that helps users write Python code.
+## Requirements:
+-you will be given a task and you will write the code to solve the task.
+-you have a dataset called **df** contains the following information:
+df.columns:{df.columns.to_list()}
+df.dtypes:{df.dtypes.to_dict()}
+
+-you have to write the code to solve the task using the dataset df.
+-you can use pandas to manipulate the dataframe.
+-you can use matplotlib to plot the data.
+-you can use seaborn to plot the data.
+-don't use print or input statements in the code.
+-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.
+-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 
+-use only the column specified in the task.
+-you have to extract the column names and the plot type  from the prompt bellow  and use them in the code.
+-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}.
         """
         generator = pipeline("text-generation", model="deepseek-ai/deepseek-coder-1.3b-instruct", device_map="auto")
         output = generator(message, max_length=1000)

requirements.txt

@@ -8,4 +8,5 @@ PyMuPDF
 matplotlib
 transformers
 seaborn
-jinja2
+jinja2
+torch

templates/index.html

@@ -6,7 +6,7 @@
   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
   <title>SmartDoc AI</title>
   <link rel="stylesheet" href="static/styles/index.css" />
-  <link rel="stylesheet" href="static/styles/all.min.css" />
+  <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.7.2/css/all.min.css" />
   <link rel="stylesheet" href="static/styles/notification.css" />
 </head>