initial commit

app.py

@@ -0,0 +1,157 @@
+import gradio as gr
+import re
+from neo4j import GraphDatabase
+
+# Static Neo4j connection details
+NEO4J_URI = "bolt://localhost:7687"
+NEO4J_USER = "neo4j"
+NEO4J_PASSWORD = "0507@#Ny"
+
+# Function to query details from Neo4j
+def query_details_from_neo4j(issue_key, driver):
+    details = {}
+    try:
+        with driver.session() as session:
+            result = session.run(
+                """
+                MATCH (i:IssueKey {key: $issue_key})
+                OPTIONAL MATCH (i)-[:HAS_SUMMARY]->(s:Summary)
+                OPTIONAL MATCH (i)-[:HAS_CREATION_DATE]->(cd:CreationDate)
+                OPTIONAL MATCH (i)-[:HAS_ASSIGNEE]->(a:Assignee)
+                OPTIONAL MATCH (i)-[:HAS_REPORTER]->(r:Reporter)
+                OPTIONAL MATCH (i)-[:HAS_REQUIREMENT_ID]->(rid:RequirementId)
+                OPTIONAL MATCH (i)-[:HAS_LABELS]->(l:Labels)
+                OPTIONAL MATCH (i)-[:HAS_COMPONENTS]->(c:Components)
+                RETURN s.text AS summary, cd.value AS creation_date, a.name AS assignee,
+                       r.name AS reporter, rid.value AS requirement_id, l.value AS labels,
+                       c.value AS components
+                """,
+                issue_key=issue_key
+            )
+            record = result.single()
+            if record:
+                details["summary"] = record["summary"]
+                details["creation_date"] = record["creation_date"]
+                details["assignee"] = record["assignee"]
+                details["reporter"] = record["reporter"]
+                details["requirement_id"] = record["requirement_id"]
+                details["labels"] = record["labels"]
+                details["components"] = record["components"]
+    except Exception as e:
+        print(f"Error querying Neo4j: {e}")
+    return details
+
+# Function to extract entities from user input
+def extract_entities_from_user_input(user_input):
+    entities = {}
+    issue_key_match = re.search(r"[A-Z]+-\d+", user_input)
+    if issue_key_match:
+        entities["issue_key"] = issue_key_match.group(0)
+    
+    requested_details = re.findall(r"(issue key|creation date|assignee|reporter|requirement id|labels|components|summary)", user_input, flags=re.IGNORECASE)
+    entities["requested_details"] = requested_details
+    
+    return entities
+
+# Function to handle the random response from the chatbot
+def random_response(driver, message, history):
+    try:
+        # Process user input
+        entities = extract_entities_from_user_input(message)
+        # Check if issue key is provided
+        issue_key = entities.get("issue_key")
+        if issue_key:
+            details = query_details_from_neo4j(issue_key, driver)
+            if details:
+                requested_details = entities.get("requested_details", [])
+                if not requested_details:
+                    bot_message = "Please specify the details you want (e.g., 'show me creation date and summary')"
+                else:
+                    bot_message = ""
+                    for detail in requested_details:
+                        if detail.lower() == "issue key":
+                            bot_message += f"Issue Key: {issue_key}\n"
+                        elif detail.lower() == "creation date":
+                            bot_message += f"Creation Date: {details['creation_date']}\n"
+                        elif detail.lower() == "assignee":
+                            bot_message += f"Assignee: {details['assignee']}\n"
+                        elif detail.lower() == "reporter":
+                            bot_message += f"Reporter: {details['reporter']}\n"
+                        elif detail.lower() == "requirement id":
+                            bot_message += f"Requirement ID: {details['requirement_id']}\n"
+                        elif detail.lower() == "labels":
+                            bot_message += f"Labels: {details['labels']}\n"
+                        elif detail.lower() == "components":
+                            bot_message += f"Components: {details['components']}\n"
+                        elif detail.lower() == "summary":
+                            bot_message += f"Summary: {details['summary']}\n"
+                        else:
+                            bot_message += f"Unknown detail: {detail}\n"
+            else:
+                bot_message = f"No details found for issue key {issue_key}"
+        else:
+            bot_message = "Issue key not found in the query"
+        
+        history.append(bot_message)
+        return history, "\n".join(history)
+    except Exception as e:
+        return history, f"Error connecting to Neo4j: {e}"
+
+# Function to handle the authentication process
+def authenticate(user, password):
+    try:
+        # Neo4j driver initialization
+        driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USER, NEO4J_PASSWORD))
+        with driver.session() as session:
+            session.run("RETURN 1")
+        return driver, True, "Authentication successful!"
+    except Exception as e:
+        return None, False, f"Authentication failed: {e}"
+
+# Function to show the chatbot page upon successful authentication
+def show_chatbot_page(auth_status, auth_message):
+    if auth_status:
+        return gr.update(visible=False), gr.update(visible=True), auth_message
+    else:
+        return gr.update(visible=True), gr.update(visible=False), auth_message
+
+# Main Gradio app
+with gr.Blocks() as demo:
+    auth_status = gr.State(False)
+    driver = gr.State(None)
+    history = gr.State([])
+
+    # Authentication Page
+    with gr.Column() as login_page:
+        user = gr.Textbox(label="Jira Username")
+        password = gr.Textbox(label="Jira Password", type="password")
+        login_button = gr.Button("Login")
+        auth_message = gr.Textbox(label="Authentication Status", interactive=False)
+
+    # Chatbot Interface
+    with gr.Column(visible=False) as chatbot_page:
+        message = gr.Textbox(label="Enter your query", lines=3)
+        output = gr.Textbox(label="Response", lines=15)
+        submit_button = gr.Button("Query")
+        clear_button = gr.Button("Clear")
+        undo_button = gr.Button("Undo")
+
+    def clear_chat(history):
+        history.clear()
+        return [], ""
+
+    def undo_last(history):
+        if history:
+            history.pop()
+        return history, "\n".join(history)
+
+    login_button.click(authenticate, inputs=[user, password], outputs=[driver, auth_status, auth_message]).then(
+        show_chatbot_page, inputs=[auth_status, auth_message], outputs=[login_page, chatbot_page, auth_message]
+    )
+
+    submit_button.click(random_response, inputs=[driver, message, history], outputs=[history, output])
+    clear_button.click(clear_chat, inputs=[history], outputs=[history, output])
+    undo_button.click(undo_last, inputs=[history], outputs=[history, output])
+
+if __name__ == "__main__":
+    demo.launch(share=True)