Update app.py

app.py

@@ -7,11 +7,34 @@ from huggingface_hub import create_repo
 #from gradio import forms
 import time
 
+#########
+import streamlit as st
+
+st.title("Drop files onto the chat input field")
+
+# Create a chat input field with file upload capabilities
+chat_input = st.chat_input(
+    placeholder="Type your message or drop a file...",
+    key="chat_input",
+    accept_drops=True,
+    on_submit=handle_file_drop
+)
+
+def handle_file_drop(files):
+    # Do something with the dropped files (e.g., process them, store them, etc.)
+    print(f"Dropped files: {files}")
+
+#if __name__ == "__main__":
+#    st.launch(function_name="app")
+#########
 ###############################
 
 import streamlit as st
 import pyautogui
 
+import os
+os.environ["DISPLAY"] = ":99.0"
+
 st.title("Chat Interface with Screenshot Capability")
 
 # Create a chat input field