Update app.py

app.py

@@ -10,6 +10,10 @@ from paddleocr import PaddleOCR, draw_ocr
 from PIL import Image
 import torch
 import re
+import matplotlib
+matplotlib.use('Agg')
+import matplotlib.pyplot as plt
+import numpy as np
 
 w_style="""
 <style>
@@ -76,6 +80,21 @@ def fx8(x,y,z):
     res=re.sub(r'{0}'.format(x),y,z)
     return res
     
+def plot(v, a):
+    g = 9.81
+    theta = a / 180 * 3.14
+    tmax = ((2 * v) * np.sin(theta)) / g
+    timemat = tmax * np.linspace(0, 1, 40)[:, None]
+
+    x = (v * timemat) * np.cos(theta)
+    y = ((v * timemat) * np.sin(theta)) - ((0.5 * g) * (timemat**2))
+
+    fig = plt.figure()
+    plt.scatter(x=x, y=y, marker=".")
+    plt.xlim(0, 100)
+    plt.ylim(0, 60)
+    return fig
+    
 with demo:
     gr.Markdown(
     """
@@ -120,7 +139,13 @@ with demo:
                     zz2_input = [gr.Textbox(placeholder='输入正则表达式，如"\d+"',label="输入正则表达式"),gr.Textbox(placeholder='正则匹配对象替换为啥',label="输入替换为的字符串"),gr.Textbox(placeholder='待替换对象',label="输入原始字符串")]
                     zz2_output = gr.Textbox(label="输出",show_label=True)
                     zz2_button = gr.Button("开始正则替换>>")
-            
+        with gr.TabItem("图表展示"):
+            with gr.Group()
+                speed = gr.Slider(1, 30, 25, label="Speed")
+                angle = gr.Slider(0, 90, 45, label="Angle")
+                output = gr.Plot()
+            tb_button = gr.Button("Run")
+         
     tj_button.click(fx1, inputs=text_input, outputs=text_output)
     cs_button.click(fx2, inputs=val_input, outputs=json_output,api_name="ghqj")
     dq_button.click(fx3, inputs=file_input, outputs=table_output,api_name="df")
@@ -129,5 +154,6 @@ with demo:
     ocr_button.click(fx6,inputs=ocr_input,outputs=ocr_output,api_name="ocr")
     zz1_button.click(fx7,inputs=zz1_input,outputs=zz1_output,api_name="zzpp")
     zz2_button.click(fx8,inputs=zz2_input,outputs=zz2_output,api_name="zzth")
+    tb_button.click(plot, [speed, angle], output)   
 
 demo.launch(enable_queue=True)