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()