update

app.py

@@ -6,6 +6,10 @@ hf_dataset = load_dataset("abishekmahi/tamil-kavithai-tanglish", split="train")
 df = hf_dataset.to_pandas()
 
 # Ensure columns are strings
+df["Content"] = df["Content"].astype(str)
+df["Title"] = df["Title"].astype(str)
+df["Category"] = df["Category"].astype(str)
+
 df["TanglishTitle"] = df["TanglishTitle"].astype(str)
 df["TanglishContent"] = df["TanglishContent"].astype(str)
 df["TanglishCategory"] = df["TanglishCategory"].astype(str)
@@ -23,12 +27,12 @@ def search_kavithai(query):
 
     kavithai = result.iloc[0]
     return f"""
-🏷️ **Title**: {kavithai['TanglishTitle']}
+🏷️ **Title**: {kavithai['Title']}
 
-📜 **Category**: {kavithai['TanglishCategory']}
+📜 **Category**: {kavithai['Category']}
 
 📝 **Kavithai**:
-{kavithai['TanglishContent']}
+{kavithai['Content']}
 """
 
 # Gradio interface
@@ -42,4 +46,4 @@ chat_interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    chat_interface.launch()
+    chat_interface.launch()