Update rag_service.py

rag_service.py

@@ -114,6 +114,16 @@ class RAGService:
             for text_element in docs_by_type["texts"]:
                 context_text += str(text_element)
     
+            # Add images only if context exists
+            if len(docs_by_type["images"]) > 0:
+                for image in docs_by_type["images"]:
+                    prompt_content.append(
+                        {
+                            "type": "image_url",
+                            "image_url": {"url": f"data:image/jpeg;base64,{image}"},
+                        }
+                    )
+            
             # Always use this flexible prompt
             prompt_template = f"""
             You are a helpful AI assistant. 
@@ -126,31 +136,19 @@ class RAGService:
             """
         
             prompt_content = [{"type": "text", "text": prompt_template}]
+            
+            return ChatPromptTemplate.from_messages([HumanMessage(content=prompt_content)])
 
         else:
-            # No context found - use direct LLM prompt
+            # Generic question with no context or images
             prompt_template = f"""
             You are a helpful AI assistant. Answer the following question using your general knowledge:
-            
             Question: {user_question}
             """
-            
-            prompt_content = [{"type": "text", "text": prompt_template}]
 
-            
-            
-        
-        # Add images only if context exists
-        if len(docs_by_type["images"]) > 0:
-            for image in docs_by_type["images"]:
-                prompt_content.append(
-                    {
-                        "type": "image_url",
-                        "image_url": {"url": f"data:image/jpeg;base64,{image}"},
-                    }
-                )
-    
-        return ChatPromptTemplate.from_messages([HumanMessage(content=prompt_content)])
+            return ChatPromptTemplate.from_messages(
+                [HumanMessage(content=prompt_template.strip())]  # plain string
+            )
 
     def ask_question(self, question: str):
         """Process a question and return response"""