update app file

app.py

@@ -15,6 +15,7 @@ pd.options.display.max_columns = 25
 pd.options.display.max_rows = 300
 
 stable_diffusion = gr.Blocks.load(name="spaces/stabilityai/stable-diffusion")
+code_generation=gr.Blocks.load(name="spaces/THUDM/CodeGeeX")
 ### ————————————————————————————————————————
 
 title="DataTeller"
@@ -28,6 +29,11 @@ def get_images(prompt):
     gallery_dir = stable_diffusion(prompt, fn_index=2)
     return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)]
 
+def get_code(prompt):
+    examples_dir = code_generation(prompt, fn_index=2)
+    return [os.path.join(examples_dir, output) for output in os.listdir(examples_dir)]
+
+
 
 def magic_whisper_to_sd(audio, guidance_scale, nb_iterations, seed):
 
@@ -42,7 +48,7 @@ def generate_code(audio, guidance_scale, nb_iterations, seed):
 
     whisper_results = translate(audio)
     prompt = whisper_results[2]
-    code = generate_code(prompt)
+    code = get_code(prompt)
 
     return code
 
@@ -333,7 +339,7 @@ with gr.Blocks(css=css) as demo:
 
         gr.Markdown(
             """
-            ## 3. Use whisper results for data visualization code generation
+            ## 3. Use whisper results for code generation
             """
         )
         with gr.Column():
@@ -397,18 +403,18 @@ with gr.Blocks(css=css) as demo:
         #                     ax.annotate('{}'.format(p.get_height()), (p.get_x(), p.get_height()+1))
         #                 return figure
 
-        #         inputs = [
-        #             gr.Dropdown(["Find Data Correlation", "Filter Correlation Data", "Business Travel vs Attrition", "Employee Experience vs Attrition", "Age vs Attrition",], label="Data Correlation and Visualization")
-        #         ]
-        #         outputs = gr.Plot()
-
-        #         demo2 = gr.Interface(
-        #             fn = outbreak,
-        #             inputs = inputs,
-        #             outputs = outputs,
-        #             title="Employee-Experience: Data Correlation and Pattern Visualization",
-        #             allow_flagging=False
-        #         )
+    #         inputs = [
+    #             gr.Dropdown(["Find Data Correlation", "Filter Correlation Data", "Business Travel vs Attrition", "Employee Experience vs Attrition", "Age vs Attrition",], label="Data Correlation and Visualization")
+    #         ]
+    #         outputs = gr.Plot()
+
+    #         demo2 = gr.Interface(
+    #             fn = outbreak,
+    #             inputs = inputs,
+    #             outputs = outputs,
+    #             title="Employee-Experience: Data Correlation and Pattern Visualization",
+    #             allow_flagging=False
+    #         )
 
 
         gr.Markdown("""