fix: save as plot, persistr

README.md

@@ -12,6 +12,9 @@ short_description: openchat for financial creditcard fraud
 ---
 
 # Financial RAG
+![Data Frame Question](./data_plot.png)
+
+![PDF Paper Question](./pdf_question.png)
 
 ## Model Selection:
 In this task i select **decoder-only** model, because we want them as conversation model. Not specific task. To avoid hallucination, we add RAG as context.
@@ -30,6 +33,9 @@ I choose `sentence-transformers/all-MiniLM-L6-v2` embedding model because it hav
 ## Ollama API
 As alternative we provie self hosted ollama api on hugginface. This API already enabled with "qwen3-4b" model. [link](https://mrfirdauss-ollama-api.hf.space)
 
+## Design Pattern
+I apply **`Factory Design Pattern`**, because we want all library we use have same method, even originally the method of the library was different. Thus we create a class as Factory Abstract Class.
+
 ## Schema
 ```
 User Input

src/FinancialAgentApp.py

@@ -23,13 +23,19 @@ class FinancialAgentApp (ABC):
     def render_header(self):
         self.st.title("Financial Agent")
 
+
     def render_messages(self):
-        """Render previous chat messages."""
+        """Render previous chat messages with roles."""
         for message in self.st.session_state.messages:
+            role = message.get("role", "assistant")  # default to assistant if missing
+
             if message.get("type") == "plot":
-                self.st.image(message["content"])
+                with self.st.chat_message(role):
+                    self.st.pyplot(message["content"])
             else:
-                self.st.markdown(message["content"])
+                with self.st.chat_message(role):
+                    self.st.markdown(message["content"])
+
 
     @abstractmethod
     def __stream_answer__(self, instructions, input_messages):
@@ -82,12 +88,15 @@ class FinancialAgentApp (ABC):
             if fig.get_axes():  # if a chart was generated
                 with self.st.chat_message("assistant"):
                     self.st.pyplot(fig)
-                plt.close(fig)
+                buf = self.__safe_savefig__()  # BytesIO PNG
+                # Add the plot as a chat message in session state
                 self.st.session_state.messages.append({
                     "role": "assistant",
                     "type": "plot",
-                    "content": self.__safe_savefig__()
+                    "content": buf
                 })
+                plt.close(fig)
+
 
             context_prompt = "## CONTEXT DATAFRAME.\n"
             context_prompt += str(local_scope.get("result", ""))