Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 from transformers import pipeline
 import PyPDF2
 import markdown
+import matplotlib.pyplot as plt
+import io
+import base64
 
 # Preload models
 models = {
@@ -19,6 +22,23 @@ def load_model(model_name):
         loaded_models[model_name] = pipeline("question-answering", model=models[model_name])
     return loaded_models[model_name]
 
+def generate_score_chart(score):
+    plt.figure(figsize=(6, 4))
+    plt.bar(["Confidence Score"], [score], color='skyblue')
+    plt.ylim(0, 1)
+    plt.ylabel("Score")
+    plt.title("Confidence Score")
+    
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close()
+    buf.seek(0)
+    return base64.b64encode(buf.getvalue()).decode()
+
+def generate_report(answer, score, score_explanation, score_chart):
+    report = f"### Answer:\n\n{answer}\n\n### Confidence Score: {score}\n\n### Score Explanation:\n\n{score_explanation}\n\n![Score Chart](data:image/png;base64,{score_chart})"
+    return report
+
 def answer_question(model_name, file, question, status):
     status = "Loading model..."
     model = load_model(model_name)
@@ -42,11 +62,17 @@ def answer_question(model_name, file, question, status):
     answer = result['answer']
     score = result['score']
     
+    # Generate the score chart
+    score_chart = generate_score_chart(score)
+    
     # Explain score
     score_explanation = f"The confidence score ranges from 0 to 1, where a higher score indicates higher confidence in the answer's correctness. In this case, the score is {score:.2f}. A score closer to 1 implies the model is very confident about the answer."
     
+    # Generate the report
+    report = generate_report(answer, f"{score:.2f}", score_explanation, score_chart)
+    
     status = "Model loaded"
-    return answer, f"{score:.2f}", score_explanation, status
+    return answer, f"{score:.2f}", score_explanation, score_chart, report, status
 
 # Define the Gradio interface
 with gr.Blocks() as interface:
@@ -73,6 +99,8 @@ with gr.Blocks() as interface:
         answer_output = gr.Textbox(label="Answer")
         score_output = gr.Textbox(label="Confidence Score")
         explanation_output = gr.Textbox(label="Score Explanation")
+        chart_output = gr.Image(label="Score Chart")
+        report_output = gr.Markdown(label="Report")
     
     with gr.Row():
         submit_button = gr.Button("Submit")
@@ -85,7 +113,7 @@ with gr.Blocks() as interface:
     submit_button.click(
         on_submit,
         inputs=[model_dropdown, file_input, question_input],
-        outputs=[answer_output, score_output, explanation_output, status_output]
+        outputs=[answer_output, score_output, explanation_output, chart_output, report_output, status_output]
     )
 
 if __name__ == "__main__":