update code

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import pipeline
-import plotly.graph_objs as go
 
 # Load the Visual QA model
 generator = pipeline("visual-question-answering", model="jihadzakki/blip1-medvqa")
@@ -9,21 +8,11 @@ def format_answer(image, question, history):
     try:
         result = generator(image, question, max_new_tokens=50)
         predicted_answer = result[0].get('answer', 'No answer found')
-        history.append(f"Question: {question} | Answer: {predicted_answer}")
-        # Create a simple chart for demonstration purposes
-        chart = create_chart(predicted_answer)
-        return f"Predicted Answer: {predicted_answer}", chart, history
-    except Exception as e:
-        return f"Error: {str(e)}", None, history
-
-def create_chart(predicted_answer):
-    # Example chart data
-    labels = ['Positive', 'Negative']
-    values = [1, 0] if predicted_answer.lower() == 'yes' else [0, 1]
-    colors = ['blue', 'red'] if predicted_answer.lower() == 'yes' else ['red', 'blue']
+        history.append((image, f"Question: {question} | Answer: {predicted_answer}"))
 
-    fig = go.Figure(data=[go.Pie(labels=labels, values=values, marker=dict(colors=colors))])
-    return fig
+        return f"Predicted Answer: {predicted_answer}", history
+    except Exception as e:
+        return f"Error: {str(e)}", history
 
 def switch_theme(mode):
     if mode == "Light Mode":
@@ -34,6 +23,12 @@ def switch_theme(mode):
 def save_feedback(feedback):
     return "Thank you for your feedback!"
 
+def display_history(history):
+    log_entries = []
+    for img, text in history:
+        log_entries.append((img, text))
+    return log_entries
+
 # Build the Visual QA application using Gradio with improvements
 with gr.Blocks(
     theme=gr.themes.Soft(
@@ -52,24 +47,22 @@ with gr.Blocks(
 
         with gr.Column():
             answer_output = gr.Textbox(label="Result Prediction")
-            chart_output = gr.Plot(label="Interactive Chart")
 
     history_state = gr.State([])  # Initialize the history state
 
     submit_button.click(
         format_answer,
         inputs=[image_input, question_input, history_state],
-        outputs=[answer_output, chart_output, history_state],
+        outputs=[answer_output, history_state],
         show_progress=True
     )
 
     with gr.Row():
-        history = gr.Textbox(label="History Log", lines=10)
-        gr.Markdown("**Log of previous interactions:**")
+        history_gallery = gr.Gallery(label="History Log", elem_id="history_log")
         submit_button.click(
-            lambda history: "\n".join(history),
+            display_history,
             inputs=[history_state],
-            outputs=[history]
+            outputs=[history_gallery]
         )
 
     with gr.Accordion("Help", open=False):
@@ -99,4 +92,4 @@ with gr.Blocks(
             outputs=[feedback_input]
         )
 
-VisualQAApp.launch(share=True, server_name="0.0.0.0", server_port=8080, debug=True)
+VisualQAApp.launch(share=True, server_name="0.0.0.0", server_port=8080, debug=True)

requirements.txt

@@ -3,3 +3,4 @@ gradio
 plotly
 torch
 tensorflow
+keras