Edit app.py

app-plain.py

@@ -890,7 +890,7 @@ def guess_date(string):
 # Main function
 def main():
     # Data loading
-    path = "./demo"
+    path = "./tmp"
     print("Searching CSV files in ", path, "...")
     # path = "/kaggle/input/artemis-test"
 

app.py

@@ -424,8 +424,6 @@ def modelTFT(csv_file, prax):
 
     #display(train.sample(4))
 
-    """<a id ="3"></a><h3 style="background:#0554f2; border:0; border-radius: 4px; color:#f5f6f7">Model Implementation in Pytorch-Forecasting </h3>"""
-
     # Add a time_idx (an sequence of consecutive integers that goes from min to max date)
 
     data = (data.merge((data[['Date/Time']].drop_duplicates(ignore_index=True)
@@ -656,8 +654,6 @@ def modelTFT_OpenGap(csv_file, prax):
 
     #display(train.sample(4))
 
-    """<a id ="3"></a><h3 style="background:#0554f2; border:0; border-radius: 4px; color:#f5f6f7">Model Implementation in Pytorch-Forecasting </h3>"""
-
     # Add a time_idx (an sequence of consecutive integers that goes from min to max date)
 
     data = (data.merge((data[['Date/Time']].drop_duplicates(ignore_index=True)
@@ -878,12 +874,6 @@ def generate_csv(data_list):
         writer.writerow(data_list)
     csv_file.close()
 
-def fileOutput():
-    today = date.today().strftime("%Y_%m_%d")
-    filename = f"result.csv"
-    shutil.copyfile(filename, f"result_{today}.csv")
-    return f"result_{today}.csv"
-
 def guess_date(string):
     for fmt in ["%Y/%m/%d", "%d-%m-%Y", "%Y%m%d", "%m/%d/%Y", "%d/%m/%Y", "%Y-%m-%d", "%d/%m/%y", "%m/%d/%y"]:
         try:
@@ -905,7 +895,7 @@ def main(files):
         prax[0] = df['Ticker'][0]
         prax[1] = df['Close'][len(df)-1]
         print('------------------')
-        df = df.drop(['EMARSI'], axis=1)
+        #df = df.drop(['EMARSI'], axis=1)
         #df['Date/Time'] = pd.to_datetime(df['Date/Time'])
         for i in range(len(df)):
             x = guess_date(df['Date/Time'][i])
@@ -932,8 +922,8 @@ def main(files):
         generate_csv(prax)
         # Reset prax
         prax = [0,0,0,0,0,0,0]
-    f1 = fileOutput()
-    return f1
+    today = date.today().strftime("%Y_%m_%d")
+    return f"result_{today}.csv"
 
 gradioApp = gr.Interface(fn=main, inputs=gr.File(file_count="multiple", file_type=".csv"), outputs="file")