Update localGPTUI/localGPTUI.py

localGPTUI/localGPTUI.py

@@ -1,72 +1,72 @@
-import argparse
-import os
-import sys
-import tempfile
-
-import requests
-from flask import Flask, render_template, request
-from werkzeug.utils import secure_filename
-
-sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
-
-app = Flask(__name__)
-app.secret_key = "LeafmanZSecretKey"
-
-API_HOST = "http://localhost:5110/api"
-
-
-# PAGES #
-@app.route("/", methods=["GET", "POST"])
-def home_page():
-    if request.method == "POST":
-        if "user_prompt" in request.form:
-            user_prompt = request.form["user_prompt"]
-            print(f"User Prompt: {user_prompt}")
-
-            main_prompt_url = f"{API_HOST}/prompt_route"
-            response = requests.post(main_prompt_url, data={"user_prompt": user_prompt})
-            print(response.status_code)  # print HTTP response status code for debugging
-            if response.status_code == 200:
-                # print(response.json())  # Print the JSON data from the response
-                return render_template("home.html", show_response_modal=True, response_dict=response.json())
-        elif "documents" in request.files:
-            delete_source_url = f"{API_HOST}/delete_source"  # URL of the /api/delete_source endpoint
-            if request.form.get("action") == "reset":
-                response = requests.get(delete_source_url)
-
-            save_document_url = f"{API_HOST}/save_document"
-            run_ingest_url = f"{API_HOST}/run_ingest"  # URL of the /api/run_ingest endpoint
-            files = request.files.getlist("documents")
-            for file in files:
-                print(file.filename)
-                filename = secure_filename(file.filename)
-                with tempfile.SpooledTemporaryFile() as f:
-                    f.write(file.read())
-                    f.seek(0)
-                    response = requests.post(save_document_url, files={"document": (filename, f)})
-                    print(response.status_code)  # print HTTP response status code for debugging
-            # Make a GET request to the /api/run_ingest endpoint
-            response = requests.get(run_ingest_url)
-            print(response.status_code)  # print HTTP response status code for debugging
-
-    # Display the form for GET request
-    return render_template(
-        "home.html",
-        show_response_modal=False,
-        response_dict={"Prompt": "None", "Answer": "None", "Sources": [("ewf", "wef")]},
-    )
-
-
-if __name__ == "__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(
-        "--host",
-        type=str,
-        default="127.0.0.1",
-        help="Host to run the UI on. Defaults to 127.0.0.1. "
-        "Set to 0.0.0.0 to make the UI externally "
-        "accessible from other devices.",
-    )
-    args = parser.parse_args()
-    app.run(debug=False, host=args.host, port=args.port)
+import argparse
+import os
+import sys
+import tempfile
+
+import requests
+from flask import Flask, render_template, request
+from werkzeug.utils import secure_filename
+
+sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
+
+app = Flask(__name__)
+app.secret_key = "LeafmanZSecretKey"
+
+API_HOST = "https://dkdaniz-katara.hf.space:5110/api"
+
+
+# PAGES #
+@app.route("/", methods=["GET", "POST"])
+def home_page():
+    if request.method == "POST":
+        if "user_prompt" in request.form:
+            user_prompt = request.form["user_prompt"]
+            print(f"User Prompt: {user_prompt}")
+
+            main_prompt_url = f"{API_HOST}/prompt_route"
+            response = requests.post(main_prompt_url, data={"user_prompt": user_prompt})
+            print(response.status_code)  # print HTTP response status code for debugging
+            if response.status_code == 200:
+                # print(response.json())  # Print the JSON data from the response
+                return render_template("home.html", show_response_modal=True, response_dict=response.json())
+        elif "documents" in request.files:
+            delete_source_url = f"{API_HOST}/delete_source"  # URL of the /api/delete_source endpoint
+            if request.form.get("action") == "reset":
+                response = requests.get(delete_source_url)
+
+            save_document_url = f"{API_HOST}/save_document"
+            run_ingest_url = f"{API_HOST}/run_ingest"  # URL of the /api/run_ingest endpoint
+            files = request.files.getlist("documents")
+            for file in files:
+                print(file.filename)
+                filename = secure_filename(file.filename)
+                with tempfile.SpooledTemporaryFile() as f:
+                    f.write(file.read())
+                    f.seek(0)
+                    response = requests.post(save_document_url, files={"document": (filename, f)})
+                    print(response.status_code)  # print HTTP response status code for debugging
+            # Make a GET request to the /api/run_ingest endpoint
+            response = requests.get(run_ingest_url)
+            print(response.status_code)  # print HTTP response status code for debugging
+
+    # Display the form for GET request
+    return render_template(
+        "home.html",
+        show_response_modal=False,
+        response_dict={"Prompt": "None", "Answer": "None", "Sources": [("ewf", "wef")]},
+    )
+
+
+if __name__ == "__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(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Host to run the UI on. Defaults to 127.0.0.1. "
+        "Set to 0.0.0.0 to make the UI externally "
+        "accessible from other devices.",
+    )
+    args = parser.parse_args()
+    app.run(debug=False, host=args.host, port=args.port)