Update app.py

app.py

@@ -2,16 +2,32 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 import requests
+from transformers import pipeline
+from sentence_transformers import SentenceTransformer, util
 
 # Hugging Face Inference Client
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# Load a pre-trained model for sentence similarity
+similarity_model = SentenceTransformer('all-mpnet-base-v2') 
+
 # Function to analyze issues and provide solutions
 def analyze_issues(issue_text, model_name):
   nlp = pipeline("text-generation", model=model_name)
   result = nlp(issue_text)
   return result[0]['generated_text']
 
+# Function to find related issues
+def find_related_issues(issue_text, issues):
+  issue_embedding = similarity_model.encode(issue_text)
+  related_issues = []
+  for issue in issues:
+    title_embedding = similarity_model.encode(issue['title'])
+    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)
+  return related_issues[:3] # Return top 3 most similar issues
+
 # Function to handle chat responses
 def respond(
     message,
@@ -71,7 +87,12 @@ def respond(
                     issue = issues[issue_number]
                     issue_text = issue['title'] + "\n\n" + issue['body']
                     resolution = analyze_issues(issue_text, "gpt2")  # Default to gpt2 for now
-                    yield f"Resolution for Issue '{issue['title']}':\n{resolution}"
+
+                    # 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}"
                 else:
                     yield f"Error fetching GitHub issues: {response.status_code}"
             except Exception as e: