Update app.py

app.py

@@ -14,47 +14,6 @@ from urllib.parse import urlparse, quote
 import webbrowser
 import spaces
 
-device = "cuda"
-
-@spaces.GPU()
-def stream_chat(
-    message: str,
-    history: list,
-    system_prompt: str,
-    temperature: float = 0.5,
-    max_new_tokens: int = 16000,
-    top_p: float = 1.0,
-    top_k: int = 15,
-    penalty: float = 0.9,
-):
-    print(f'message: {message}')
-    print(f'history: {history}')
-
-    conversation = [
-        {"role": "system", "content": system_prompt}
-    ]
-    for prompt, answer in history:
-        conversation.extend([
-            {"role": "user", "content": prompt},
-            {"role": "assistant", "content": answer},
-        ])
-
-    conversation.append({"role": "user", "content": message})
-
-    input_ids = tokenizer.apply_chat_template(conversation, add_generation_prompt=True, return_tensors="pt").to(model.device)
-
-    streamer = TextIteratorStreamer(tokenizer, timeout=60.0, skip_prompt=True, skip_special_tokens=True)
-
-    generate_kwargs = dict(
-        input_ids=input_ids,
-        max_new_tokens = max_new_tokens,
-        do_sample = False if temperature == 0 else True,
-        top_p = top_p,
-        top_k = top_k,
-        temperature = temperature,
-        eos_token_id=[128001,128008,128009],
-        streamer=streamer,
-    )
 # Constants
 INPUT_DIRECTORY = 'input'
 OUTPUT_DIRECTORY = 'output'
@@ -81,13 +40,6 @@ def initialize_environment():
     for directory in directories:
         os.makedirs(directory, exist_ok=True)
 
-# Check if constants are defined
-def check_constants():
-    constants = ['LOGS_DIRECTORY', 'RESOLUTIONS_DIRECTORY', 'REPOS_DIRECTORY', 'INPUT_DIRECTORY', 'OUTPUT_DIRECTORY']
-    for constant in constants:
-        if constant not in globals():
-            raise NameError(f"Constant '{constant}' is not defined")
-
 # GitHub API handler
 class GitHubAPI:
     def __init__(self, token: str):
@@ -401,12 +353,12 @@ custom_html = """
             return;
         }
         try {
-            const response = await fetch('/fetch_issues', {
+            const response = await fetch('/fetch-issues', {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
                 },
-                body: JSON.stringify({ token: token, repoUrl: repoUrl }),
+                body: JSON.stringify({ githubToken: token, repoUrl: repoUrl }),
             });
             if (!response.ok) {
                 throw new Error(`HTTP error! status: ${response.status}`);
@@ -439,12 +391,12 @@ custom_html = """
             return;
         }
         try {
-            const response = await fetch('/resolve_issue', {
+            const response = await fetch('/resolve-issue', {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
                 },
-                body: JSON.stringify({ token: token, repoUrl: repoUrl, issueNumber: issueNumber, resolution: resolution, forkedRepoUrl: forkedRepoUrl }),
+                body: JSON.stringify({ githubToken: token, repoUrl: repoUrl, issueNumber: issueNumber, resolution: resolution, forkedRepoUrl: forkedRepoUrl }),
             });
             if (!response.ok) {
                 throw new Error(`HTTP error! status: ${response.status}`);
@@ -453,7 +405,7 @@ custom_html = """
             if (data.error) {
                 outputTextarea.value = data.error;
             } else {
-                outputTextarea.value = data.result;
+                outputTextarea.value = data.output;
             }
         } catch (error) {
             outputTextarea.value = `Error resolving issue: ${error.message}`;
@@ -467,7 +419,7 @@ custom_html = """
             return;
         }
         try {
-            const response = await fetch('/extract_info', {
+            const response = await fetch('/extract-info', {
                 method: 'POST',
                 headers: {
                     'Content-Type': 'application/json',
@@ -514,18 +466,20 @@ def signal_handler(signum, frame):
     cleanup()
     sys.exit(0)
 
-if __name__ == "__main__":
-    # Check if constants are defined
-    check_constants()
-
+@spaces.GPU()
+def main():
     # Initialize environment and logger
     initialize_environment()
-    logger = initialize_logger()  # Initialize logger globally
+    logger = initialize_logger()
 
     # Register cleanup handlers
     atexit.register(cleanup)
     signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
 
-    # Create and launch the Gradio interface
+    # Create Gradio interface
     demo = create_gradio_interface()
-    demo.launch()
+    demo.launch()
+
+if __name__ == "__main__":
+    main()