Update frontend.py

frontend.py

@@ -13,6 +13,7 @@ app = Flask(__name__)
 app.secret_key = "LeafmanZSecretKey"
 
 API_HOST = "https://dkdaniz-katara.hf.space:5110/api"
+    
 
 # PAGES #
 @app.route("/", methods=["GET", "POST"])
@@ -54,9 +55,11 @@ def home_page():
         show_response_modal=False,
         response_dict={"Prompt": "None", "Answer": "None", "Sources": [("ewf", "wef")]},
     )
+    
 
 
-if __name__ == "__main__":
+
+def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--port", type=int, default=5111, help="Port to run the UI on. Defaults to 5111.")
     parser.add_argument(
@@ -68,4 +71,7 @@ if __name__ == "__main__":
         "accessible from other devices.",
     )
     args = parser.parse_args()
-    app.run(debug=False, host=args.host, port=args.port)
+    app.run(debug=False, host=args.host, port=args.port)
+    
+if __name__ == "__main__":
+    main()