Update app.py

app.py

@@ -76,6 +76,52 @@ def home_page():
     </div>
   </div>
     """
+
+def format_cyber_response(user_question: str, raw_answer: str, model="gpt-4") -> str:
+    """
+    Uses an AI model to rephrase a raw cyber answer into a professionally structured response
+    according to the style and tone of the user question.
+    """
+
+    prompt = f"""
+You are an expert technical writer and cybersecurity analyst.
+
+Your task is to take the following user question and raw answer related to MITRE ATT&CK framework 
+or cyberattack scenario generation, and rewrite the answer professionally based on the nature 
+and tone of the question.
+
+Instructions:
+1. Analyze the question to detect whether it is asking for an explanation, summary, technical analysis, or use-case scenario.
+2. Rewrite the raw answer to match the expected response style (educational, technical, formal, or analytical).
+3. Use domain-appropriate terminology.
+4. Make the output coherent, structured, and professionally written.
+
+---
+
+Question:
+{user_question}
+
+Raw Answer:
+{raw_answer}
+
+---
+
+Final Response:
+"""
+
+    return prompt
+from gradio_client import Client
+
+def    ask_wasm(text):
+        txt=format_cyber_response(text)
+        client = Client("wasmdashai/LAHJA-AI")
+        result = client.predict(
+        		prompt=txt,
+        		api_name="/generate_from_prompt"
+        )
+        return result
+
+
 def t2t(text, namn_model):
     if namn_model == "Group":
         out = ASGAI.Group.predictAPI(text)
@@ -83,7 +129,9 @@ def t2t(text, namn_model):
         out = ASGAI.Tec.predictAPI(text)
     else:
         out = ASGAI.Soft.predictAPI(text)
-    return str(out)
+
+    anwai=ask_wasm(text,str(out))
+    return  anwai
 
 def t2seq(text, namn_model):
     if namn_model == "Group":
@@ -92,6 +140,9 @@ def t2seq(text, namn_model):
         out = ASGAI.Tec.Predict_ALL(text)
     else:
         out = ASGAI.Soft.Predict_ALL(text)
+
+
+    
     return str(out)
 def echo(message, history):
     text=t2seq(message,"Group")