Update app.py

app.py

@@ -8,7 +8,7 @@ from datetime import datetime
 from sklearn.metrics import mean_absolute_error, mean_squared_error
 from statsmodels.tsa.statespace.sarimax import SARIMAX
 from statsmodels.tsa.holtwinters import ExponentialSmoothing
-
+from openpyxl.utils import get_column_letter
 from openpyxl import Workbook
 from openpyxl.drawing.image import Image
 from pandas import DataFrame
@@ -36,6 +36,29 @@ def forecast(name,duration):
   })
   return forecast_df
 
+def changewidth(worksheet):
+    column_widths = {}
+    
+    # Calculate the maximum width for each column
+    for column in worksheet.columns:
+        column_letter = get_column_letter(column[0].column)
+        max_length = 0
+        for cell in column:
+            try:
+                if len(str(cell.value)) > max_length:
+                    max_length = len(str(cell.value))
+            except TypeError:
+                pass
+        column_widths[column_letter] = max_length
+
+    # Set the column widths based on the maximum width for each column
+    for column_letter, width in column_widths.items():
+        column_width = (width + 2) * 1.2  # Adjust the width by adding some padding
+        worksheet.column_dimensions[column_letter].width = column_width
+
+    return worksheet
+
+
 warnings.filterwarnings('ignore')
 def predict(operation,file):
   if file == None:
@@ -50,7 +73,7 @@ def predict(operation,file):
 
     # output_name = "Forecasted.csv"
     # predicted.to_csv(output_name,index=False)
-    c = 0
+
     workbook = Workbook()
     # Remove the default sheet
     default_sheet = workbook.active
@@ -73,6 +96,7 @@ def predict(operation,file):
       # Insert the plot image into the Excel file
       img = Image(plot_filename)
       worksheet.add_image(img, 'D1')
+      worksheet = changewidth(worksheet)
     # Save the Excel file
     now = datetime.now()
     formatted_datetime = now.strftime("%H:%M %d-%m-%Y")
@@ -89,5 +113,7 @@ iface = gr.Interface(fn=predict,
                      
                      inputs=[gr.Radio(label='Predict ahead:',choices=['Month','Week'],value='Month'),gr.File(label="CSV file")],
 
-                     outputs=[gr.File(label="CSV file"),gr.Textbox(label='Log',interactive=False)])
+                     outputs=[gr.File(label="CSV file"),gr.Textbox(label='Log',interactive=False)],
+
+                     debug=True)
 iface.launch()