troubleshoot chatui

app/main.py

@@ -169,6 +169,72 @@ async def root():
         }
     }
 
+# Add this to your FastAPI app after the existing routes
+
+@app.post("/v1/chat/completions")
+async def openai_compatible_endpoint(request: Request):
+    """OpenAI-compatible endpoint that ChatUI might prefer"""
+    try:
+        data = await request.json()
+        logger.info(f"OpenAI endpoint received: {data}")
+        
+        # Extract messages from OpenAI format
+        messages = data.get("messages", [])
+        user_message = ""
+        
+        for msg in reversed(messages):
+            if msg.get("role") == "user":
+                user_message = msg.get("content", "")
+                break
+        
+        if not user_message:
+            return {"error": "No user message found"}
+        
+        result = process_chatfed_query_core(query=user_message)
+        
+        # Return in OpenAI format
+        return {
+            "choices": [{
+                "message": {
+                    "role": "assistant", 
+                    "content": result.result
+                },
+                "finish_reason": "stop"
+            }],
+            "model": "chatfed",
+            "usage": {"total_tokens": 0}
+        }
+        
+    except Exception as e:
+        logger.error(f"Error in OpenAI endpoint: {str(e)}")
+        return {"error": str(e)}
+
+# Also add a simple text endpoint
+@app.post("/simple-chat")
+async def simple_chat_endpoint(request: Request):
+    """Simple endpoint that accepts any text input"""
+    try:
+        data = await request.json()
+        logger.info(f"Simple chat received: {data}")
+        
+        # Try to extract text from various possible fields
+        text = (data.get("text") or 
+                data.get("query") or 
+                data.get("prompt") or 
+                data.get("input") or 
+                str(data))
+        
+        result = process_chatfed_query_core(query=text)
+        
+        return {
+            "response": result.result,
+            "metadata": result.metadata
+        }
+        
+    except Exception as e:
+        logger.error(f"Error in simple chat: {str(e)}")
+        return {"error": str(e)}
+
 # Mount Gradio at a specific path instead of root to avoid conflicts
 demo = create_gradio_interface()
 app.mount("/ui", demo)