Fix: Correct dataset field mapping for question/answer extraction

app_gradio.py

@@ -75,7 +75,8 @@ class TextilindoAI:
         similarities = []
         
         for example in self.dataset:
-            question = example.get('question', '').lower()
+            # Try different field names that might exist in the dataset
+            question = example.get('question', example.get('instruction', '')).lower()
             similarity = SequenceMatcher(None, query.lower(), question).ratio()
             similarities.append((similarity, example))
         
@@ -92,8 +93,9 @@ class TextilindoAI:
             # Build context from similar examples
             context = ""
             for example in similar_examples:
-                question = example.get('question', '')
-                answer = example.get('answer', '')
+                # Try different field names that might exist in the dataset
+                question = example.get('question', example.get('instruction', ''))
+                answer = example.get('answer', example.get('output', ''))
                 context += f"Q: {question}\nA: {answer}\n\n"
             
             # Create prompt
@@ -109,8 +111,11 @@ Please provide a helpful response based on the context above. If the context doe
             # For now, return a simple response based on context
             if similar_examples:
                 # Return the most similar answer
-                best_answer = similar_examples[0].get('answer', '')
-                return f"Based on our knowledge base: {best_answer}"
+                best_answer = similar_examples[0].get('answer', similar_examples[0].get('output', ''))
+                if best_answer:
+                    return f"Based on our knowledge base: {best_answer}"
+                else:
+                    return "I found some relevant information but couldn't extract a proper answer. Please try rephrasing your question."
             else:
                 return "I'm sorry, I don't have specific information about that. Please contact Textilindo directly for more details."