Upload app.py

app.py

@@ -18,7 +18,7 @@ import matplotlib.pyplot as plt
 import io
 
 # BASE_URL = os.getenv("BASE_URL")
-API_KEY = os.getenv("OPENAI_API_KEY")
+API_KEY = os.getenv("API_KEY")
 
 BASE_URL = "https://api.openai.com"
 print(f"BASE_URL: {BASE_URL}")
@@ -615,18 +615,27 @@ def create_interface(users, chosen_image, reference_code):
     return demo
 
 
-if __name__ == "__main__":
-    users = Path("users.txt").read_text().splitlines()
-    users = set(user.strip() for user in users if user.strip())
-    chosen_image = pick_random_image()
-    reference_code = chosen_image.replace(".png", ".py")
-    # code_context = extract_code_context(reference_code)
-    demo = create_interface(users, chosen_image, reference_code)
+# if __name__ == "__main__":
+#     users = Path("users.txt").read_text().splitlines()
+#     users = set(user.strip() for user in users if user.strip())
+#     chosen_image = pick_random_image()
+#     reference_code = chosen_image.replace(".png", ".py")
+#     # code_context = extract_code_context(reference_code)
+#     demo = create_interface(users, chosen_image, reference_code)
 
-    # demo.launch(
-    #     server_name=args.server_name,
-    #     server_port=args.server_port,
-    #     share=args.share,
-    # )
+#     # demo.launch(
+#     #     server_name=args.server_name,
+#     #     server_port=args.server_port,
+#     #     share=args.share,
+#     # )
 
-    demo.launch()
+#     demo.launch()
+
+users = Path("users.txt").read_text().splitlines()
+users = set(user.strip() for user in users if user.strip())
+chosen_image = pick_random_image()
+reference_code = chosen_image.replace(".png", ".py")
+# code_context = extract_code_context(reference_code)
+demo = create_interface(users, chosen_image, reference_code)
+
+demo.launch()