NitinBot001 commited on
Commit
3e949dd
·
verified ·
1 Parent(s): 362683e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -7
app.py CHANGED
@@ -29,7 +29,7 @@ def download_model():
29
  def start_tunnel():
30
  # Start nport tunnel
31
  tunnel_process = subprocess.Popen(
32
- ["npx", "nport", "-s", "ai-service-new", "-p", "5000"],
33
  stdout=subprocess.PIPE,
34
  stderr=subprocess.PIPE,
35
  )
@@ -57,6 +57,11 @@ def push_tunnel_url_to_repo(tunnel_url):
57
  # Clone the repository
58
  repo_dir = "/tmp/repo"
59
  repo_url = f"https://x-access-token:{GH_PAT}@github.com/NitinBot001/Audio-url-new-js.git"
 
 
 
 
 
60
  subprocess.run(
61
  ["git", "clone", repo_url, repo_dir],
62
  check=True,
@@ -70,19 +75,27 @@ def push_tunnel_url_to_repo(tunnel_url):
70
  subprocess.run(["git", "config", "user.email", "[email protected]"], check=True)
71
  subprocess.run(["git", "config", "user.name", "github-actions"], check=True)
72
 
73
- # Commit and push changes
74
  subprocess.run(["git", "add", "instance.json"], check=True)
75
- subprocess.run(["git", "commit", "-m", f"Update tunnel URL to {tunnel_url}"], check=True)
76
- subprocess.run(["git", "push", "origin", "main"], check=True)
77
 
78
- @app.route("/chat", methods=["POST"])
 
 
 
 
 
 
 
 
79
  def chat():
80
- data = request.json
 
 
81
  # Construct the prompt without duplicate special tokens
82
  prompt = (
83
  f"<|begin_of_text|>"
84
  f"<|start_header_id|>user<|end_header_id|>\n"
85
- f"{data.get('message', '')}"
86
  f"<|eot_id|>\n"
87
  f"<|start_header_id|>assistant<|end_header_id|>\n"
88
  )
 
29
  def start_tunnel():
30
  # Start nport tunnel
31
  tunnel_process = subprocess.Popen(
32
+ ["npx", "nport", "-s", "ai-service", "-p", "5000"],
33
  stdout=subprocess.PIPE,
34
  stderr=subprocess.PIPE,
35
  )
 
57
  # Clone the repository
58
  repo_dir = "/tmp/repo"
59
  repo_url = f"https://x-access-token:{GH_PAT}@github.com/NitinBot001/Audio-url-new-js.git"
60
+
61
+ # Remove the existing repo directory if it exists
62
+ if os.path.exists(repo_dir):
63
+ subprocess.run(["rm", "-rf", repo_dir], check=True)
64
+
65
  subprocess.run(
66
  ["git", "clone", repo_url, repo_dir],
67
  check=True,
 
75
  subprocess.run(["git", "config", "user.email", "[email protected]"], check=True)
76
  subprocess.run(["git", "config", "user.name", "github-actions"], check=True)
77
 
78
+ # Add the file to the staging area
79
  subprocess.run(["git", "add", "instance.json"], check=True)
 
 
80
 
81
+ # Check if there are changes to commit
82
+ result = subprocess.run(["git", "status", "--porcelain"], stdout=subprocess.PIPE, text=True)
83
+ if result.stdout.strip(): # If there are changes
84
+ subprocess.run(["git", "commit", "-m", f"Update tunnel URL to {tunnel_url}"], check=True)
85
+ subprocess.run(["git", "push", "origin", "main"], check=True)
86
+ else:
87
+ print("No changes to commit.")
88
+
89
+ @app.route("/chat", methods=["GET"])
90
  def chat():
91
+ # Get the message from the query parameter
92
+ message = request.args.get("message", "")
93
+
94
  # Construct the prompt without duplicate special tokens
95
  prompt = (
96
  f"<|begin_of_text|>"
97
  f"<|start_header_id|>user<|end_header_id|>\n"
98
+ f"{message}"
99
  f"<|eot_id|>\n"
100
  f"<|start_header_id|>assistant<|end_header_id|>\n"
101
  )