Update app.py

app.py

@@ -115,7 +115,7 @@ class DetailedExplainer:
         )
         result = self.explainer(
             prompt,
-            max_length=200,
+            max_new_tokens=800,
             min_length=80,
             do_sample=False
         )
@@ -125,8 +125,15 @@ class DetailedExplainer:
         concepts = self.extract_concepts(text)
         explanations = {}
         for concept in concepts:
-            explanations[concept] = self.explain_concept(concept, context)
-        return {"concepts": concepts, "explanations": explanations}
+            try:
+                explanations[concept] = self.explain_concept(concept, context)
+            except Exception as e:
+                explanations[concept] = f"[Error explaining concept: {e}]"
+        return {
+            "answer_sentence": text.strip(),
+            "concepts": concepts,
+            "explanations": explanations
+        }
 
 class AdvancedPDFAnalyzer:
     def __init__(self):
@@ -231,14 +238,25 @@ def ask_question_gradio(question: str):
         result = analyzer.answer_question(question, documents)
         answer = result['answer']
         confidence = result['confidence']
-        explanation = "\n\n".join(
-            f"🔹 {concept}: {desc}"
-            for concept, desc in result.get("explanations", {}).get("explanations", {}).items()
-        )
-        return f"📌 **Answer**: {answer}\n\n🔒 **Confidence**: {confidence:.2f}\n\n📘 **Explanations**:\n{explanation}"
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
 
+        explanation_data = result.get("explanations", {})
+        answer_sentence = explanation_data.get("answer_sentence", answer)
+        explanations = explanation_data.get("explanations", {})
+
+        markdown_output = f"\U0001F4CC **Answer:**\n{answer_sentence}\n\n"
+        markdown_output += f"\U0001F512 **Confidence:** {confidence:.2f}\n\n"
+        markdown_output += f"\U0001F4D8 **Concept Explanations:**\n"
+
+        if not explanations:
+            markdown_output += "\nNo key concepts found to explain."
+        else:
+            for concept, desc in explanations.items():
+                markdown_output += f"\n\U0001F539 **{concept}**\n{desc}\n"
+
+        return markdown_output
+
+    except Exception as e:
+        return f"\u274C Error: {str(e)}"
 
 demo = gr.Interface(
     fn=ask_question_gradio,
@@ -248,4 +266,4 @@ demo = gr.Interface(
     description="Ask a question based on the document loaded in this system."
 )
 
-demo.launch()
+demo.launch()