iisadia commited on
Commit
019bd86
Β·
verified Β·
1 Parent(s): ef7d92e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -62,6 +62,10 @@ def analyze_requirement(requirement):
62
  domain_prompt = f"Classify the domain for the following requirement in one word (e.g., E-commerce, Education, etc.):\n\n{requirement}\n\nDomain:"
63
  domain = call_mistral_api(domain_prompt).strip()
64
 
 
 
 
 
65
  # Use Groq for defect analysis and rewriting
66
  defects_prompt = f"""List ONLY the major defects in the following requirement (e.g., Ambiguity, Incompleteness, etc.) in 1-2 words each:\n\n{requirement}\n\nDefects:"""
67
  defects = call_groq_api(defects_prompt).strip()
@@ -73,6 +77,7 @@ def analyze_requirement(requirement):
73
  "Requirement": requirement,
74
  "Type": req_type,
75
  "Domain": domain,
 
76
  "Defects": defects,
77
  "Rewritten": rewritten
78
  }
@@ -235,6 +240,9 @@ def main():
235
  <div class="analysis-badge" style="background: #e8eaf6; color: #303f9f;">
236
  🏷️ <b>Domain:</b> {result['Domain']}
237
  </div>
 
 
 
238
  <div class="analysis-badge" style="background: #b39ddb; color: purple;">
239
  πŸ”Ž <b>Identified Defects:</b><br> Explanation: {result['Defects']}
240
  </div>
 
62
  domain_prompt = f"Classify the domain for the following requirement in one word (e.g., E-commerce, Education, etc.):\n\n{requirement}\n\nDomain:"
63
  domain = call_mistral_api(domain_prompt).strip()
64
 
65
+ stakeholder_prompt = f"""Identify key major stakeholders for this requirement. Provide a comma-separated list of roles.
66
+ Examples: Users, Administrators, Developers, Security Team. Requirement:\n\n{requirement}\n\nStakeholders:"""
67
+ stakeholders = call_groq_api(stakeholder_prompt).strip()
68
+
69
  # Use Groq for defect analysis and rewriting
70
  defects_prompt = f"""List ONLY the major defects in the following requirement (e.g., Ambiguity, Incompleteness, etc.) in 1-2 words each:\n\n{requirement}\n\nDefects:"""
71
  defects = call_groq_api(defects_prompt).strip()
 
77
  "Requirement": requirement,
78
  "Type": req_type,
79
  "Domain": domain,
80
+ "Stakeholders": stakeholders,
81
  "Defects": defects,
82
  "Rewritten": rewritten
83
  }
 
240
  <div class="analysis-badge" style="background: #e8eaf6; color: #303f9f;">
241
  🏷️ <b>Domain:</b> {result['Domain']}
242
  </div>
243
+ <div class="analysis-badge" style="background: #e0f2f1; color: #00796b;">
244
+ πŸ‘₯ <b>Stakeholders:</b> {result['Stakeholders']}
245
+ </div>
246
  <div class="analysis-badge" style="background: #b39ddb; color: purple;">
247
  πŸ”Ž <b>Identified Defects:</b><br> Explanation: {result['Defects']}
248
  </div>