Update app.py

app.py

@@ -118,6 +118,18 @@ class Network(object):
 
       # return idx, other_data_input, true_output, predicted_output
       return predicted_output, true_output
+      
+  
+  def synthesis_from_spectrum(self, other_data_input):
+      # Get the input
+      data_input = other_data_input.reshape((1, 3*self.F))
+ 
+      # Get the outputs
+      predicted_output = self.network.predict(data_input)
+      predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
+
+      # return idx, other_data_input, true_output, predicted_output
+      return true_output
 
   def get_geometry(self, idx=None):
 
@@ -235,9 +247,26 @@ with gradio.Blocks() as synthesis_demo:
     btn1.click(fn=performance, inputs=[num], outputs=[perf])
     btn2.click(fn=simple_synthesis, inputs=[num], outputs=[pred, true])
     
-all_synthesis_demos = gradio.TabbedInterface([synthesis_demo], ["Random Spectrum from Data"])
+    
+with gradio.Blocks() as synthesis_demo2:
+    with gradio.Row():
+        perf = gradio.Timeseries(x="Frequency", y=['Surge', 'Heave', 'Pitch'], label="Performance")
+
+    with gradio.Row():
+        btn2 = gradio.Button("Synthesize Geometry")
+
+    with gradio.Row():
+      with gradio.Column():
+          pred = gradio.Plot(label="Predicted")
+
+      with gradio.Column():
+          true = gradio.Plot(label="True")
+
+    btn2.click(fn=synthesis_from_spectrum, inputs=[num], outputs=[pred, true])
+    
+all_synthesis_demos = gradio.TabbedInterface([synthesis_demo, synthesis_demo2], ["Spectrum from Data", "Spectrum from File"])
 
-all_analysis_demos = gradio.TabbedInterface([analysis_demo], ["Random Geometry from Data"])
+all_analysis_demos = gradio.TabbedInterface([analysis_demo], ["Geometry from Data"])
 
 demo = gradio.TabbedInterface([all_analysis_demos, all_synthesis_demos], ["Analysis", "Synthesis"])
 demo.launch()