Update app.py

app.py

@@ -4,6 +4,10 @@ import os
 import requests
 from transformers import pipeline
 from sentence_transformers import SentenceTransformer, util
+import logging
+
+# Enable detailed logging
+logging.basicConfig(level=logging.INFO)
 
 # Hugging Face Inference Client
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -23,6 +27,7 @@ def analyze_issues(issue_text: str, model_name: str, severity: str = None, progr
     Returns:
     str: The analyzed issue and solution.
     """
+    logging.info(f"Analyzing issue: {issue_text} with model: {model_name}")
     prompt = f"""Issue: {issue_text}
 Severity: {severity}
 Programming Language: {programming_language}
@@ -50,9 +55,12 @@ Please provide a comprehensive resolution in the following format:
 """
     try:
         nlp = pipeline("text-generation", model=model_name, max_length=1000)  # Increase max_length
+        logging.info(f"Pipeline created with model: {model_name}")
         result = nlp(prompt)
+        logging.info(f"Model output: {result}")
         return result[0]['generated_text']
     except Exception as e:
+        logging.error(f"Error analyzing issue with model {model_name}: {e}")
         return f"Error analyzing issue with model {model_name}: {e}"
 
 ### Function to find related issues
@@ -65,6 +73,7 @@ def find_related_issues(issue_text: str, issues: list) -> list:
     Returns:
     list: The list of related issues.
     """
+    logging.info(f"Finding related issues for: {issue_text}")
     issue_embedding = similarity_model.encode(issue_text)
     related_issues = []
     for issue in issues:
@@ -72,6 +81,7 @@ def find_related_issues(issue_text: str, issues: list) -> list:
         similarity = util.cos_sim(issue_embedding, title_embedding)[0][0]
         related_issues.append((issue, similarity))
     related_issues = sorted(related_issues, key=lambda x: x[1], reverse=True)
+    logging.info(f"Found related issues: {related_issues}")
     return related_issues[:3]  # Return top 3 most similar issues
 
 ### Function to fetch GitHub issues
@@ -85,6 +95,7 @@ def fetch_github_issues(github_api_token: str, github_username: str, github_repo
     Returns:
     list: The list of GitHub issues.
     """
+    logging.info(f"Fetching GitHub issues for: {github_username}/{github_repository}")
     url = f"https://api.github.com/repos/{github_username}/{github_repository}/issues"
     headers = {
         "Authorization": f"Bearer {github_api_token}",
@@ -92,8 +103,11 @@ def fetch_github_issues(github_api_token: str, github_username: str, github_repo
     }
     response = requests.get(url, headers=headers)
     if response.status_code == 200:
-        return response.json()
+        issues = response.json()
+        logging.info(f"Fetched issues: {issues}")
+        return issues
     else:
+        logging.error(f"Error fetching issues: {response.status_code}")
         raise Exception(f"Error fetching issues: {response.status_code}")
 
 ### Function to handle chat responses
@@ -136,12 +150,17 @@ def respond(
     issues = []
 
     messages = [{"role": "system", "content": system_message}]
+    logging.info(f"System message: {system_message}")
+
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+            logging.info(f"User message: {user_msg}")
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+            logging.info(f"Assistant message: {assistant_msg}")
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    logging.info(f"Command received: {command}")
 
     if command == "/github":
         if not github_api_token:
@@ -152,6 +171,7 @@ def respond(
                 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":
@@ -178,6 +198,7 @@ def respond(
 
             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"):
@@ -189,8 +210,13 @@ def respond(
             prompt = f"Generate code for the following: {code_description}\nProgramming Language: {programming_language}"
             try:
                 generated_code = analyze_issues(prompt, selected_model)  # Reuse analyze_issues for code generation
-                yield f"```{programming_language}\n{generated_code}\n```"
-            except Exception as e:
+                yield f"
+
+
+{programming_language}\n{generated_code}\n
+
+except Exception as e:
+                logging.error(f"Error generating code: {e}")
                 yield f"Error generating code: {e}"
 
     elif command.startswith("/explain_concept"):
@@ -203,6 +229,7 @@ def respond(
                 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"):
@@ -215,6 +242,7 @@ def respond(
                 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"):
@@ -223,27 +251,42 @@ def respond(
             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```"
-            try:
+            prompt = f"Translate the following code to {target_language}:\n
+
+
+\n{code}\n
+
+try:
                 translated_code = analyze_issues(prompt, selected_model)
-                yield f"```{target_language}\n{translated_code}\n```"
-            except Exception as e:
+                yield f"
+
+
+{target_language}\n{translated_code}\n
+
+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 = ""
-        for message in client.chat_completion(
-            messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
-        ):
-            token = message.choices[0].delta.content
-            response += token
-            yield 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():
@@ -322,4 +365,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860, show_header=False)
+    demo.queue().launch(share=True, server_name="0.0.0.0", server_port=7860, show_header=False, debug=True)