Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from typing import List, Tuple
 from transformers import pipeline
+import matplotlib.pyplot as plt
 
 # Constants
 WIDTH, HEIGHT = 600, 600
@@ -15,7 +16,11 @@ velocities: List[Tuple[float, float]] = [(0.0, 0.0)] * len(touch_points)
 is_affected: List[bool] = [False] * len(touch_points)
 
 # Set up the Hugging Face pipeline
-text_generator = pipeline('text-generation', model='gpt2')
+@st.cache_resource
+def load_model():
+    return pipeline('text-generation', model='gpt2')
+
+text_generator = load_model()
 
 # Streamlit app
 st.title("Artificial Touch Simulation")
@@ -56,27 +61,33 @@ def on_tap(x, y):
             is_affected[i] = True
     
     # Generate a description of the touch
-    with touch_container:
-        st.write(f"Touch at ({x:.2f}, {y:.2f})")
-        text = text_generator(f"The user touched the screen at ({x:.2f}, {y:.2f}).", max_length=100, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=1)[0]['generated_text']
-        st.write(text)
+    st.write(f"Touch at ({x:.2f}, {y:.2f})")
+    text = text_generator(f"The user touched the screen at ({x:.2f}, {y:.2f}).", max_length=100, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=1)[0]['generated_text']
+    st.write(text)
     
     update_points()
 
-while True:
-    with touch_container:
-        for i, (x, y) in enumerate(touch_points):
-            if is_affected[i]:
-                st.circle((x, y), 5, color="red", fill_opacity=0.5)
-            else:
-                st.circle((x, y), 5, color="navy", fill_opacity=0.5)
-        
-        if st.button("Tap the screen"):
-            on_tap(st.session_state.get('x', 0), st.session_state.get('y', 0))
-        
-        st.experimental_set_query_params(x=touch_points[0][0], y=touch_points[0][1])
-        st.session_state['x'] = st.experimental_get_query_params().get('x', [0])[0]
-        st.session_state['y'] = st.experimental_get_query_params().get('y', [0])[0]
-        
-    update_points()
-    st.experimental_rerun()
+# Initialize session state
+if 'x' not in st.session_state:
+    st.session_state.x = 0
+if 'y' not in st.session_state:
+    st.session_state.y = 0
+
+# Main app logic
+fig, ax = plt.subplots(figsize=(6, 6))
+ax.set_xlim(0, WIDTH)
+ax.set_ylim(0, HEIGHT)
+
+for i, (x, y) in enumerate(touch_points):
+    color = "red" if is_affected[i] else "navy"
+    ax.add_artist(plt.Circle((x, y), 5, color=color, alpha=0.5))
+
+touch_container.pyplot(fig)
+
+if touch_container.button("Tap the screen"):
+    on_tap(st.session_state.x, st.session_state.y)
+
+st.session_state.x = st.slider("X coordinate", 0, WIDTH, st.session_state.x)
+st.session_state.y = st.slider("Y coordinate", 0, HEIGHT, st.session_state.y)
+
+update_points()