Update app.py

app.py

@@ -163,133 +163,125 @@ def respond(
     logging.info(f"Command received: {command}")
 
     if command == "/github":
-        if not github_api_token:
-            yield "Please enter your GitHub API token first. [Click here to get your token](https://github.com/settings/tokens)"
-        else:
-            try:
-                issues = fetch_github_issues(github_api_token, github_username, github_repository)
-                issue_list = "\n".join([f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
-                yield f"Available GitHub Issues:\n{issue_list}\n\nEnter the issue number to analyze:"
-            except Exception as e:
-                logging.error(f"Error fetching GitHub issues: {e}")
-                yield f"Error fetching GitHub issues: {e}"
-
-    elif command == "/help":
-        help_message = f"""Available commands:
-              - `/github`: Analyze a GitHub issue
-              - `/help`: Show this help message
-              - `/generate_code [code description]`: Generate code based on the description
-              - `/explain_concept [concept]`: Explain a concept
-              - `/write_documentation [topic]`: Write documentation for a given topic
-              - `/translate_code [code] to [target language]`: Translate code to another language"""
-        yield help_message  # Yield the pre-formatted help message
-
-    elif command.isdigit() and issues:
+    if not github_api_token:
+        yield "Please enter your GitHub API token first. <https://github.com/settings/tokens>"
+    else:
+        try:
+            issues = fetch_github_issues(github_api_token, github_username, github_repository)
+            issue_list = "\n".join([f"{i+1}. {issue['title']}" for i, issue in enumerate(issues)])
+            yield f"Available GitHub Issues:\n{issue_list}\n\nEnter the issue number to analyze:"
+        except Exception as e:
+            logging.error(f"Error fetching GitHub issues: {e}")
+            yield f"Error fetching GitHub issues: {e}"
+
+elif command == "/help":
+    help_message = f"""Available commands:
+          - `/github`: Analyze a GitHub issue
+          - `/help`: Show this help message
+          - `/generate_code [code description]`: Generate code based on the description
+          - `/explain_concept [concept]`: Explain a concept
+          - `/write_documentation [topic]`: Write documentation for a given topic
+          - `/translate_code [code] to [target language]`: Translate code to another language"""
+    yield help_message  # Yield the pre-formatted help message
+
+elif command.isdigit() and issues:
+    try:
+        issue_number = int(command) - 1
+        issue = issues[issue_number]
+        issue_text = issue['title'] + "\n\n" + issue['body']
+        resolution = analyze_issues(issue_text, selected_model, severity, programming_language)
+
+        # Find and display related issues
+        related_issues = find_related_issues(issue_text, issues)
+        related_issue_text = "\n".join(
+            [f"- {issue['title']} (Similarity: {similarity:.2f})" for issue, similarity in related_issues]
+        )
+
+        yield f"Resolution for Issue '{issue['title']}':\n{resolution}\n\nRelated Issues:\n{related_issue_text}"
+    except Exception as e:
+        logging.error(f"Error analyzing issue: {e}")
+        yield f"Error analyzing issue: {e}"
+
+elif command.startswith("/generate_code"):
+    # Extract the code description from the command
+    code_description = command.replace("/generate_code", "").strip()
+    if not code_description:
+        yield "Please provide a description of the code you want to generate."
+    else:
+        prompt = f"Generate code for the following: {code_description}\nProgramming Language: {programming_language}"
         try:
-            issue_number = int(command) - 1
-            issue = issues[issue_number]
-            issue_text = issue['title'] + "\n\n" + issue['body']
-            resolution = analyze_issues(issue_text, selected_model, severity, programming_language)
-
-            # Find and display related issues
-            related_issues = find_related_issues(issue_text, issues)
-            related_issue_text = "\n".join(
-                [f"- {issue['title']} (Similarity: {similarity:.2f})" for issue, similarity in related_issues]
-            )
-
-            yield f"Resolution for Issue '{issue['title']}':\n{resolution}\n\nRelated Issues:\n{related_issue_text}"
+            generated_code = analyze_issues(prompt, selected_model)
+            code_output = f"<pre>{programming_language}\n{generated_code}</pre>"
+            yield code_output  # Yield the formatted string
         except Exception as e:
-            logging.error(f"Error analyzing issue: {e}")
-            yield f"Error analyzing issue: {e}"
-
-    elif command.startswith("/generate_code"):
-        # Extract the code description from the command
-        code_description = command.replace("/generate_code", "").strip()
-        if not code_description:
-            yield "Please provide a description of the code you want to generate."
-        else:
-            prompt = f"Generate code for the following: {code_description}\nProgramming Language: {programming_language}"
-            try:
-                generated_code = analyze_issues(prompt, selected_model)
-                code_output = f"
-
-
-{programming_language}\n{generated_code}\n
-
-yield code_output  # Yield the formatted string
-            except Exception as e:
-                logging.error(f"Error generating code: {e}")
-                yield f"Error generating code: {e}"
-
-    elif command.startswith("/explain_concept"):
-        concept = command.replace("/explain_concept", "").strip()
-        if not concept:
-            yield "Please provide a concept to explain."
-        else:
-            prompt = f"Explain the concept of {concept} in detail."
-            try:
-                explanation = analyze_issues(prompt, selected_model)  # Reuse analyze_issues for explanation
-                yield explanation
-            except Exception as e:
-                logging.error(f"Error explaining concept: {e}")
-                yield f"Error explaining concept: {e}"
-
-    elif command.startswith("/write_documentation"):
-        topic = command.replace("/write_documentation", "").strip()
-        if not topic:
-            yield "Please provide a topic for documentation."
-        else:
-            prompt = f"Write comprehensive documentation for the following topic: {topic}"
-            try:
-                documentation = analyze_issues(prompt, selected_model)
-                yield documentation
-            except Exception as e:
-                logging.error(f"Error writing documentation: {e}")
-                yield f"Error writing documentation: {e}"
-
-    elif command.startswith("/translate_code"):
-        parts = command.replace("/translate_code", "").strip().split(" to ")
-        if len(parts) != 2:
-            yield "Invalid command format. Use: /translate_code [code] to [target language]"
-        else:
-            code, target_language = parts
-            prompt = f"Translate the following code to {target_language}:\n
-
-
-\n{code}\n
+            logging.error(f"Error generating code: {e}")
+            yield f"Error generating code: {e}"
 
-try:
-                translated_code = analyze_issues(prompt, selected_model)
-                code_output = f"
+elif command.startswith("/explain_concept"):
+    concept = command.replace("/explain_concept", "").strip()
+    if not concept:
+        yield "Please provide a concept to explain."
+    else:
+        prompt = f"Explain the concept of {concept} in detail."
+        try:
+            explanation = analyze_issues(prompt, selected_model)
+            yield f"<pre>{explanation}</pre>"
+        except Exception as e:
+            logging.error(f"Error explaining concept: {e}")
+            yield f"Error explaining concept: {e}"
 
+elif command.startswith("/write_documentation"):
+    topic = command.replace("/write_documentation", "").strip()
+    if not topic:
+        yield "Please provide a topic for documentation."
+    else:
+        prompt = f"Write comprehensive documentation for the following topic: {topic}"
+        try:
+            documentation = analyze_issues(prompt, selected_model)
+            yield f"<pre>{documentation}</pre>"
+        except Exception as e:
+            logging.error(f"Error writing documentation: {e}")
+            yield f"Error writing documentation: {e}"
 
-{target_language}\n{translated_code}\n
+elif command.startswith("/translate_code"):
+    parts = command.replace("/translate_code", "").strip().split(" to ")
+    if len(parts) != 2:
+        yield "Invalid command format. Use: /translate_code [code] to [target language]"
+    else:
+        code, target_language = parts
+        prompt = f"Translate the following code to {target_language}:\n\n
 
-yield code_output  # Yield the formatted string
-            except Exception as e:
-                logging.error(f"Error translating code: {e}")
-                yield f"Error translating code: {e}"
 
-    else:
-        messages.append({"role": "user", "content": command})
-        logging.info(f"User message: {command}")
+{code}\n
 
-        response = ""
-        try:
-            for message in client.chat_completion(
-                messages,
-                max_tokens=max_tokens,
-                stream=True,
-                temperature=temperature,
-                top_p=top_p,
-            ):
-                logging.info(f"Received message from chat completion: {message}")
-                token = message.choices[0].delta.content
-                response += token
-                yield response
-        except Exception as e:
-            logging.error(f"Error during chat completion: {e}")
-            yield f"An error occurred: {e}"
+try:
+translated_code = analyze_issues(prompt, selected_model)
+code_output = f"<pre>{target_language}\n{translated_code}</pre>"
+yield code_output # Yield the formatted string
+except Exception as e:
+logging.error(f"Error translating code: {e}")
+yield f"Error translating code: {e}"
+
+else:
+messages.append({"role": "user", "content": command})
+logging.info(f"User message: {command}")
+
+response = ""
+try:
+    for message in client.chat_completion(
+        messages,
+        max_tokens=max_tokens,
+        stream=True,
+        temperature=temperature,
+        top_p=top_p,
+    ):
+        logging.info(f"Received message from chat completion: {message}")
+        token = message.choices[0].delta.content
+        response += token
+        yield response
+except Exception as e:
+    logging.error(f"Error during chat completion: {e}")
+    yield f"An error occurred: {e}"
 
 with gr.Blocks() as demo:
     with gr.Row():