Add drive cycle reply feature

Author: Saransh-cpp

Experiments/Experiment.py

@@ -10,6 +10,11 @@
 foo = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(foo)
 
+spec1 = importlib.util.spec_from_file_location(
+    "PlotGraph.py", "Randomness/PlotGraph.py"
+)
+foo1 = importlib.util.module_from_spec(spec1)
+spec1.loader.exec_module(foo1)
 
 def experiment_func(cycle=None):
 
@@ -116,10 +121,13 @@ def US06_experiment_cycle():
     return cycleC
 
 
-def US06_experiment():
+def US06_experiment(reply=False):
     model = pybamm.lithium_ion.DFN()
     # import drive cycle from file
-    drive_cycle = pd.read_csv("US06.csv", comment="#", header=None).to_numpy()
+    if reply: 
+        drive_cycle = pd.read_csv("drive_cycle.csv", comment="#", header=None).to_numpy()
+    elif not reply:
+        drive_cycle = pd.read_csv("US06.csv", comment="#", header=None).to_numpy()
     # create interpolant
     param = model.default_parameter_values
     timescale = param.evaluate(model.timescale)
@@ -134,6 +142,10 @@ def US06_experiment():
     )
     sol_US06_1 = sim_US06_1.solve()
 
+    if reply:
+        time = foo1.plot_graph(sol_US06_1, sim_US06_1, reply=reply)
+        return time
+
     solved = False
     print("REACHED")
     while not solved:

Randomness/RandomPlotGenerator.py

@@ -159,6 +159,11 @@ def random_plot_generator(
                 return time, feasible
 
         elif choice == 2:
+
+            if reply:
+                time = foo2.US06_experiment(reply=reply)
+                return time
+
             repeat = True
             while repeat:
                 try:

ReplyTweet.py

@@ -3,6 +3,7 @@
 import matplotlib.pyplot as plt
 import importlib.util
 import os
+import gdown
 
 spec = importlib.util.spec_from_file_location(
     "RandomPlotGenerator.py.py", "Randomness/RandomPlotGenerator.py"
@@ -82,6 +83,15 @@ def reply_to_tweet():
                             choice=1, cycle=experiment
                         )
                         print("yes")
+                    if "drive cycle" in singleMention.full_text.lower():
+                        tweet = singleMention.full_text.split()
+                        link = tweet[-1]
+                        output = 'drive_cycle.csv'
+                        gdown.download(link, output, quiet=False)
+                        time = foo.random_plot_generator(
+                            choice=2
+                        )
+
 
             if feasible:
                 media = api.media_upload("replyFoo.png")
@@ -109,6 +119,7 @@ def reply_to_tweet():
                 )
 
             os.remove("replyFoo.png")
+            os.remove("drive_cycle.csv")
             plt.clf()
         else:
             api.update_status(

lastSeenId.txt

@@ -1 +1 @@
-1375171411245899776
+1386344261877002240